汇总到矢量输出

时间:2019-07-09 16:58:46

标签: r vector dplyr purrr summarize

假设我有以下(简化的)小标题,其中包含一组和向量中的值:

set.seed(1)
(tb_vec <- tibble(group = factor(rep(c("A","B"), c(2,3))),
             values = replicate(5, sample(3), simplify = FALSE)))
# A tibble: 5 x 2
  group values   
  <fct> <list>   
1 A     <int [3]>
2 A     <int [3]>
3 B     <int [3]>
4 B     <int [3]>
5 B     <int [3]>

tb_vec[[1,2]]
[1] 1 3 2

我想通过将每个值向量相加(向量化)来总结它们,并尝试以下操作:

tb_vec %>% group_by(group) %>% 
  summarize(vec_sum = colSums(purrr::reduce(values, rbind)))
  

错误:列vec_sum的长度必须为1(汇总值),而不是3

该错误使我感到惊讶,因为小标题(输出格式)也可以包含矢量。

我的预期输出将是以下摘要:

# A tibble: 2 x 2
  group vec_sum  
  <fct> <list>   
1 A     <dbl [3]>
2 B     <dbl [3]>

是否有一个tidyverse解决方案可以容纳summary的向量输出?我要避免分裂小标题,因为那样的话我就放松了这个因素。

1 个答案:

答案 0 :(得分:3)

您只需要在解决方案的list(.)内添加summarise,就能拥有一个包含2个元素的列,其中每个元素都是3个值的向量:

library(tidyverse)

set.seed(1)
(tb_vec <- tibble(group = factor(rep(c("A","B"), c(2,3))),
                  values = replicate(5, sample(3), simplify = FALSE)))

tb_vec %>% 
  group_by(group) %>%                              
  summarize(vec_sum = list(colSums(purrr::reduce(values, rbind)))) -> res

res$vec_sum

# [[1]]
# [1] 2 4 6
# 
# [[2]]
# [1] 6 5 7