字符串作为函数参数 r

Question

我正在使用 dplyr 并创建代码来计算用 ggplot 绘制的新数据。

我想用这段代码创建一个函数。它应该采用由 dplyr 操作的数据框列的名称。但是，尝试使用列名不起作用。请考虑下面的最小示例：

df <- data.frame(A = seq(-5, 5, 1), B = seq(0,10,1))

library(dplyr)
foo <- function (x) {
         df %>%
            filter(x < 1)
}

foo(B)

Error in filter_impl(.data, dots(...), environment()) : 
  object 'B' not found 

有什么办法可以将列名作为函数参数使用吗？

Answer 1

如果你想创建一个接受字符串“B”作为参数的函数（如你的问题标题）

foo_string <- function (x) {
         eval(substitute(df %>% filter(xx < 1),list(xx=as.name(x))))
}
foo_string("B")

如果你想创建一个接受捕获 B 作为参数的函数（如在 dplyr 中）

foo_nse <- function (x) {
         # capture the argument without evaluating it
         x <- substitute(x)
         eval(substitute(df %>% filter(xx < 1),list(xx=x)))
}
foo_nse(B)

您可以在Advanced R找到更多信息

编辑

dplyr 在 0.3 版中让事情变得更容易。带有后缀“_”的函数接受字符串或表达式作为参数

 foo_string <- function (x) {
             # construct the string
             string <- paste(x,"< 1")
             # use filter_ instead of filter
             df %>% filter_(string)
    }
foo_string("B")
 foo_nse <- function (x) {
             # capture the argument without evaluating it
             x <- substitute(x)
             # construct the expression
             expression <- lazyeval::interp(quote(xx < 1), xx = x)
             # use filter_ instead of filter
             df %>% filter_(expression)
    }
foo_nse(B)

你可以在这个vignette找到更多信息

Answer 2

我记得@Richard Scriven 回答了一个类似的问题。我认为你需要写这样的东西。

foo <- function(x,...)filter(x,...) 

@Richard Scriven 提到的是你需要在这里使用...。如果您键入 ?dplyr，您将能够找到：filter(.data, ...) 我认为您将 .data 替换为 x 或其他内容。如果你想在你的df中选择B中值小于1的行，它会是这样的。

foo <- function (x,...) filter(x,...)
foo(df, B < 1)