Tidyeval 函数中的列名列表

Question

我正在尝试创建一个将列名列表传递给 dplyr 函数的函数。如果列名称列表以 ... 形式给出，我知道如何执行此操作，如 tidyeval 文档中所述：

df <- tibble(
  g1 = c(1, 1, 2, 2, 2),
  g2 = c(1, 2, 1, 2, 1),
  a = sample(5), 
  b = sample(5)
)

my_summarise <- function(df, ...) {
  group_var <- quos(...)

  df %>%
    group_by(!!!group_var) %>%
    summarise(a = mean(a))
}

my_summarise(df, g1, g2)

但是如果我想将列名列为函数的参数，上述解决方案将不起作用（当然）：

my_summarise <- function(df, group_var, sum_var) {
  group_var <- quos(group_var) # nor enquo(group_var)
  sum_var <- enquo(sum_var)

  df %>%
    group_by(!!!group_var) %>%
    summarise(a = mean(a))
}

my_summarise(df, list(g1, g2), a)
my_summarise(df, list(g1, g2), b)

如何让列表中的项目单独报价？

这个问题和Passing dataframe column names in a function inside another function类似，但是在评论中建议使用字符串，而这里我想使用裸列名称。

Answer 1

您可以使用 alist 而不是 list 传递参数列表，因为它不会计算参数。

my_summarise = function(df, group_var, sum_var) {
    group_var = quos(!!! group_var)
    sum_var = enquo(sum_var)

    df %>%
        group_by(!!! group_var) %>%
        summarise(!! quo_name( sum_var) := mean( !! sum_var) )
}

my_summarise(df, alist(g1, g2), b)

# A tibble: 4 x 3
# Groups:   g1 [?]
     g1    g2     b
  <dbl> <dbl> <dbl>
1     1     1   2.0
2     1     2   3.0
3     2     1   4.5
4     2     2   1.0

另一种选择是直接使用 quos 传递该参数而不是 list，如所示，这样可以一起绕过一些复杂问题。

my_summarise = function(df, group_var, sum_var) {
    # group_var = quos(!!! group_var)
    sum_var = enquo(sum_var)

    df %>%
        group_by(!!! group_var) %>%
        summarise(!! quo_name( sum_var) := mean( !! sum_var) )
}

my_summarise(df, quos(g1, g2), b)

# A tibble: 4 x 3
# Groups:   g1 [?]
     g1    g2     b
  <dbl> <dbl> <dbl>
1     1     1   2.0
2     1     2   3.0
3     2     1   4.5
4     2     2   1.0

Answer 2

library(dplyr)

df <- tibble(
  g1 = c(1, 1, 2, 2, 2),
  g2 = c(1, 2, 1, 2, 1),
  a = sample(5), 
  b = sample(5)
)

my_summarise = function(df, group_var, fun_name) {

  df %>%
    group_by(!!! group_var) %>%
    summarize_all(fun_name)
}

my_summarise(df, alist(g1, g2), mean)

alist() 将参数 'g1' 和 'g2' 作为函数参数处理（不计算它们）而 !!! （与 UQS() 取消引号和拼接列表相同。sum_var 不是必需的，因为看起来您想取 'a' 和 'b' 的平均值。此外，您可以将其概括也通过传递函数。

Tidyeval 函数中的列名列表

Tidyeval with list of column names in a function

r

dplyr

tidyeval