Question

我有多个具有相同列数的数据框。

iris1 <- iris
iris2 <- iris

然后，我要提取一些特定的列，并用具有特定列的列覆盖原始数据框。

func <- function(df) {
    temp <- df %>%
    select("Species",starts_with("Sepal"))
    assign(deparse(substitute(df)),temp,envir=.GlobalEnv)
    }

如果仅将功能应用于一个数据帧，则效果很好：

func(iris1)

str(iris1)
'data.frame':   150 obs. of  3 variables:
 $ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 $ Sepal.Length: num  5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
 $ Sepal.Width : num  3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...

但是，一旦我尝试将其应用于多个数据框，它将无法正常工作：

func(list(iris1,iris2))
Error: Variable context not set

我试图找到解决方案，但大多数建议都建议使用lapply，它以列表格式返回结果。

lapply(list(iris1,iris2),func) -> result

我只想通过函数覆盖数据帧iris1和iris2，但如何呢？目前，我正在按数据帧运行函数，但我希望通过一次操作来完成。

func(iris1)
func(iris2)

Answer 1

尝试一下：

func <- function(...) {
  require(dplyr)
  mc <- match.call(expand.dots = FALSE)
  lapply(mc$..., function(n) {
    assign(deparse(n), get(deparse(n)) %>% select("Species",starts_with("Sepal")), envir = .GlobalEnv)
  })
  invisible()
}

> str(iris1)
'data.frame':   150 obs. of  5 variables:
 $ Sepal.Length: num  5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
 $ Sepal.Width : num  3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
 $ Petal.Length: num  1.4 1.4 1.3 1.5 1.4 1.7 1.4 1.5 1.4 1.5 ...
 $ Petal.Width : num  0.2 0.2 0.2 0.2 0.2 0.4 0.3 0.2 0.2 0.1 ...
 $ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
> str(iris2)
'data.frame':   150 obs. of  5 variables:
 $ Sepal.Length: num  5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
 $ Sepal.Width : num  3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
 $ Petal.Length: num  1.4 1.4 1.3 1.5 1.4 1.7 1.4 1.5 1.4 1.5 ...
 $ Petal.Width : num  0.2 0.2 0.2 0.2 0.2 0.4 0.3 0.2 0.2 0.1 ...
 $ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
> func(iris1, iris2)
> str(iris1)
'data.frame':   150 obs. of  3 variables:
 $ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 $ Sepal.Length: num  5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
 $ Sepal.Width : num  3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
> str(iris2)
'data.frame':   150 obs. of  3 variables:
 $ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 $ Sepal.Length: num  5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
 $ Sepal.Width : num  3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...

Answer 2

一种方法是使用eval，parse和paste0在for循环内运行命令。这可行，但并不理想。

for (df in c("iris1", "iris2")) {
  eval(parse(text = paste0(df, ' <- select(', df ,', "Species", starts_with("Sepal"))')))
}

这会循环遍历名称列表并为每个名称运行一个命令，因此列表c("iris1", "iris2")将运行：

iris1 <- select(iris1, "Species", starts_with("Sepal"))
iris2 <- select(iris2, "Species", starts_with("Sepal"))

如何使用assign（deparse（substitute（df）））将相同的函数应用于多个数据帧以覆盖输入变量？ [R]

2 个答案: