r - Dplyr 编写一个以列名作为输入的函数

Question

我正在编写一个要在 dplyr 中的多个列上使用的函数，但我无法将列名作为输入传递给 dplyr 的函数。

这是我想做的一个例子:

df<-tbl_df(data.frame(group=rep(c("A", "B"), each=3), var1=sample(1:100, 6), var2=sample(1:100, 6)))


example<-function(colname){
  df %>%
    group_by(group)%>%
    summarize(output=mean(sqrt(colname)))%>%
    select(output)
}
example("var1")

输出应该是这样的

df %>%
  group_by(group)%>%
  summarize(output=mean(sqrt(var1)))%>%
  select(output)

我发现了一些类似的问题，但没有任何问题可以直接应用于我的问题，因此不胜感激。我已经尝试了一些涉及 eval 的解决方案，但老实说我不知道​​我应该传递给 eval 的到底是什么。

Answer 1

这是您所期望的吗？

df<-tbl_df(data.frame(group=rep(c("A", "B"), each=3), var1=sample(1:100, 6), var2=sample(1:100, 6)))

example<-function(colname){
  df %>%
    group_by(group)%>%
    summarize(output=mean(sqrt(colname)))%>%
    select(output)
}
example( quote(var1) )
#-----
Source: local data frame [2 x 1]

    output
1 7.185935
2 8.090866