如何在R中的函数内使用dplyr/magrittr的管道？

Question

我正在尝试编写一个函数,它将数据帧和函数的名称作为参数.当我尝试写与标准的R语法功能,我可以使用取得了良好的效果eval,并substitute在建议报告由@hadley http://adv-r.had.co.nz/Computing-on-the-language.html

> df <- data.frame(y = 1:10)
> f <- function(data, x) {
+   out <- mean(eval(expr = substitute(x), envir = data))
+   return(out)
+ }
> f(data = df, x = y)
[1] 5.5

现在,当我尝试使用%>%运算符编写相同的函数时,它不起作用:

> df <- data.frame(y = 1:10)
> f <- function(data, x) {
+   data %>% 
+     eval(expr = substitute(x), envir = .) %>% 
+     mean()
+ }
> f(data = df, x = y)
Show Traceback
Rerun with Debug
 Error in eval(expr, envir, enclos) : objet 'y' introuvable 
> 

我如何使用组合管道操作员和使用eval和substitute？这对我来说似乎很棘手.

Answer 1

解决方法是

f <- function(data, x) {
  v <- substitute(x)
  data %>% 
    eval(expr = v, envir = .) %>%
    mean()
}

问题是管道函数(%>%)正在创建另一个级别的闭包,这会干扰评估substitute(x).你可以看到这个例子的不同之处

df <- data.frame(y = 1:10)
f1 <- function(data, x) {
  print(environment())
  eval(expr = environment(), envir = data)
}

f2 <- function(data, x) {
  print(environment())
  data %>% 
    eval(expr = environment(), envir = .)
}
f1(data = df, x = y)
# <environment: 0x0000000006388638>
# <environment: 0x0000000006388638>
f2(data = df, x = y)
# <environment: 0x000000000638a4a8>
# <environment: 0x0000000005f91ae0>

请注意matrittr版本中的环境有何不同.你想在substitute非标准评估的情况下尽快处理这些事情.

我希望你的用例比你的例子更复杂,因为它看起来像

mean(df$y)

将是一个更容易阅读的代码.

Answer 2

我一直在努力理解我的问题。

首先，我用该summarise()函数编写了我想要的内容：

> library(dplyr)
> df <- data.frame(y = 1:10)
> summarise_(.data = df, mean = ~mean(y))
  mean
1  5.5

然后我尝试编写自己的函数。我找到了一个似乎适用于这篇文章lazyeval中的包的解决方案。我使用和函数来编写我想要的内容。lazy()interp()

第一种可能性在这里：

> library(lazyeval)
> f <- function(data, col) {
+   col <- lazy(col)
+   inter <- interp(~mean(x), x = col)
+   summarise_(.data = data, mean = inter)    
+   }
> f(data = df, col = y)
  mean
1  5.5

我还可以使用管道：

> f <- function(data, col) {
+   col <- lazy(col)
+   inter <- interp(~mean(x), x = col)
+   data %>% 
+     summarise_(.data = ., mean = inter)    
+ }
> 
> f(data = df, col = y)
  mean
1  5.5