使用 invoke_map 将变量名作为参数传递

Question

我想使用invoke_map 来调用函数列表。我有一组变量名，我想将它们用作每个函数的参数。最终变量名将与group_by 一起使用。

这是一个例子：

library(dplyr)
library(purrr)
first_fun <- function(...){
  by_group = quos(...)
  mtcars %>%
    group_by(!!!by_group) %>%
    count()
}

second_fun <- function(...){
  by_group = quos(...)
  mtcars %>%
    group_by(!!!by_group) %>%
    summarise(avg_wt = mean(wt))
}

first_fun(mpg, cyl) # works
second_fun(mpg, cyl) # works

both_funs <- list(first_fun, second_fun)

both_funs %>%
  invoke_map(mpg, cyl) # What do I do here?

我尝试了各种尝试将变量名放在引号中，enquo 他们，使用vars，引用.data$mpg，等等，但我有点在黑暗中刺伤。

Answer 1

问题不在于您使用的是点，而在于您使用的是名称，并且当调用 map2_impl 时，这些参数会被评估。

试试这个并探索环境：

debugonce(map2)
both_funs %>% invoke_map("mpg", "cyl")

另一方面，这是可行的：

first_fun2 <- function(...){
  mtcars %>%
  {do.call(group_by_,list(.,unlist(list(...))))} %>%
    count()
}

second_fun2 <- function(...){
  mtcars %>%
  {do.call(group_by_,list(.,unlist(list(...))))} %>%
    summarise(avg_wt = mean(wt))
}

both_funs2 <- list(first_fun2, second_fun2)
both_funs2 %>% invoke_map("mpg", "cyl") 

# [[1]]
# # A tibble: 25 x 2
# # Groups:   mpg [25]
# mpg     n
# <dbl> <int>
#   1  10.4     2
# 2  13.3     1
# 3  14.3     1
# 4  14.7     1
# 5  15.0     1
# 6  15.2     2
# 7  15.5     1
# 8  15.8     1
# 9  16.4     1
# 10  17.3     1
# # ... with 15 more rows
# 
# [[2]]
# # A tibble: 25 x 2
# mpg avg_wt
# <dbl>  <dbl>
#   1  10.4 5.3370
# 2  13.3 3.8400
# 3  14.3 3.5700
# 4  14.7 5.3450
# 5  15.0 3.5700
# 6  15.2 3.6075
# 7  15.5 3.5200
# 8  15.8 3.1700
# 9  16.4 4.0700
# 10  17.3 3.7300
# # ... with 15 more rows