2

这是新的 dplyr,即将发布 Real Soon Now。

dplyr编程小插图给出了一个group_by使用外部函数中指定的分组变量进行调用的示例:

my_summarise <- function(df, group_var) {
  df %>%
    group_by(!!group_var) %>%
    summarise(a = mean(a))
}

这在提供单个分组变量时有效。但是,它因多个变量而失败。

简化示例:

f <- function(x)
{
    group_by(mtcars, !!x)
}

## works
g1 <- "cyl"
f(g1)

## doesn't work
#Error in mutate_impl(.data, dots) : 
#  Column `c("cyl", "gear")` must be length 32 (the number of rows) or one, not 2
g2 <- c("cyl", "gear")
f(g2)

在 rlang 框架内,我该如何解决这个问题?

理想情况下,我希望f' 的签名保持不变,即我将分组变量指定为单个向量,而不是通过...参数。

4

1 回答 1

3

有一个非常相似的问题:Programming with dplyr using string as input。我只是稍微修改了答案以使用symsand !!!

library(rlang)
f <- function(x){
  group_by(mtcars, !!!syms(x))
}

f(c("cyl")) %>% summarise(n())
# A tibble: 3 x 2
    cyl `n()`
  <dbl> <int>
1     4    11
2     6     7
3     8    14

f(c("cyl", "gear")) %>% summarise(n())
# A tibble: 8 x 3
# Groups:   cyl [?]
    cyl  gear `n()`
  <dbl> <dbl> <int>
1     4     3     1
2     4     4     8
3     4     5     2
4     6     3     2
5     6     4     4
6     6     5     1
7     8     3    12
8     8     5     2
于 2017-05-24T23:23:08.337 回答