r - 使用 purrr 映射按列对迭代汇总

Question

我有一个大型数据集，我希望从中获得一列按其他两列分组时的汇总估计值(平均值、中位数、计数等)。

非常努力地尝试找出如何使用 purrr 来做到这一点 - 希望让这个工作流为 future 的项目点击......但非常卡住。

作为一个可重现的例子，这适用于按 am 和 vs 分组，并估计 mpg 的汇总值

library(tidyverse)
library(rlang)

mtcars %>%
  group_by(am, vs) %>%
  summarise(mean_mpg = mean(mpg),
            median_mpg = median(mpg),
            count = n())

但是，为了扩展这个例子，假设我想为 am 和 vs 分组；然后是am和gear；然后是 am 和 carb。直觉上，这似乎是 map 应该处理的事情。

group_vars <- c("vs", "gear", "carb")
group_syms <- rlang::syms(group_vars)
sym_am <- rlang::sym("am")

mtcars %>%
  map_df(~group_by(!!sym_am, !!!group_syms) %>%
           summarise(mean_mpg = mean(mpg),
           summarise(median_mpg = median(mpg),
           summarise(count = n())
  )

#Error in !sym_am : invalid argument type

Answer 1

我们可以使用 purrr 中的 map2 将多个符号用作参数，然后在 group_by 和 summarise 输出

library(tidyverse)
map2_df(list(sym_am), group_syms, ~ mtcars %>%
         group_by(!!.x, !!.y) %>% 
         summarise(mean_mgp = mean(mpg), median_mpg = median(mpg),count = n()))