函数名称作为字符串传递的 Dplyr 非标准评估

Question

在使用dplyr 管道时，我想使用NSE 将函数传递给mutate，函数名称是从向量传递的。

示例

给定两个函数名的向量：

funs <- c("sum", "mean")

我想使用第一个值来获得总和：

require(dplyr)
mtcars %>% 
  group_by(cyl) %>% 
  mutate_(res = funs[1](hp))

这会导致错误：

Error in as.lazy_dots(list(...)) : attempt to apply non-function

do.call

基于do.call 的解决方案似乎为 sum 生成了一些结果：

mtcars %>% 
  group_by(cyl) %>% 
  mutate_(res = do.call(funs[1], .))

但尝试使用mean时失败：

>> mtcars %>% 
+   group_by(cyl) %>% 
+   mutate_(res = do.call(funs[2], .))
Error in mean.default(mpg = c(21, 21, 22.8, 21.4, 18.7, 18.1, 14.3, 24.4,  : 
  argument "x" is missing, with no default

我猜它在这里的应用方式根本没有意义。因此我的问题是：如何在dplyr 中使用nse，以便函数可以作为字符串从向量传递？

Answer 1

这两个都使用mutate而不是mutate_

mtcars %>% 
  group_by(cyl) %>% 
  mutate(res = do.call(funs[2], list(hp)))

mtcars %>% 
  group_by(cyl) %>% 
  mutate(res = match.fun(funs[2])(hp))

另外请注意，如果我们使用 [[2]] 代替 [2]，那么这些将适用于问题中出现的字符向量 funs 以及 funs <- c(sum, mean)。

Answer 2

我们可以使用get 和get 提取单个字符串的值。这里是一个函数，所以返回函数本身。

mtcars %>% 
     group_by(cyl) %>% 
     mutate(res= get(funs[1])(hp))

用于传递额外的参数

mtcars$hp[1] <- NA
mtcars %>%
      group_by(cyl) %>% 
      mutate(res= get(funs[1])(hp, na.rm = TRUE))