r - 在R中汇总组均值时如何创建有条件的新组

我有一些要总结分组均值的数据。然后，我想将一些较小的组(匹配某个n
这是一个使用tibble和dyplr的示例。

# preps
library(tibble)
library(dplyr)
set.seed(7)

# generate 4 groups with more observations
tbl_1  <- tibble(group = rep(sample(letters[1:4], 150, TRUE), each = 4),
                 score = sample(0:10, size = 600, replace = TRUE))

# generate 3 groups with less observations
tbl_2 <- tibble(group = rep(sample(letters[5:7], 50, TRUE), each = 3),
                score = sample(0:10, size = 150, replace = TRUE))

# put them into one data frame
tbl <- rbind(tbl_1, tbl_2)

# aggregate the mean scores and count the observations for each group
tbl_agg1 <- tbl %>%
  group_by(group) %>%
  summarize(MeanScore = mean(score),
            n = n())

到目前为止很容易。
接下来，我只想显示具有100多个观察值的组。所有其他组应合并为一个称为“其他”的组。

# First, calculate summary stats for groups less then n < 100
tbl_agg2 <- tbl_agg1 %>%
   filter(n<100) %>%
      summarize(MeanScore = weighted.mean(MeanScore, n),
                sumN = sum(n))

注意:上面的计算中有一个错误，现在可以纠正(@Frank:感谢您发现它!)

# Second, delete groups less then n < 100 from the aggregate table and add a row containing the summary statistics calculated above instead
tbl_agg1 <- tbl_agg1 %>%
   filter(n>100) %>%
      add_row(group = "others", MeanScore = tbl_agg2[["MeanScore"]], n = tbl_agg2[["sumN"]])

tbl_agg1基本上显示了我想要显示的内容，但是我想知道是否有更流畅，更有效的方法来执行此操作。同时，我想知道data.table方法将如何处理当前的问题。

我欢迎任何建议。

最佳答案

我猜您对“其他”组的计算是错误的。

tbl_agg1 %>% {bind_rows(
   filter(., n>100),
   filter(., n<100) %>%
   summarize(group = "other", MeanScore = weighted.mean(MeanScore, n), n = sum(n))
)}

但是，可以通过使用其他分组变量使事情从一开始就简单得多:

tbl %>%
  group_by(group) %>%
  group_by(g = replace(group, n() < 100, "other")) %>%
  summarise(n = n(), m = mean(score))

# A tibble: 5 x 3
  g         n     m
  <chr> <int> <dbl>
1 a       136  4.79
2 b       188  4.49
3 c       160  5.32
4 d       116  4.78
5 other   150  5.42

或与data.table

library(data.table)
DT = data.table(tbl)
DT[, n := .N, by=group]
DT[, .(.N, m = mean(score)), keyby=.(g = replace(group, n < 100, "other"))]

       g   N        m
1:     a 136 4.786765
2:     b 188 4.489362
3:     c 160 5.325000
4:     d 116 4.784483
5: other 150 5.420000

关于r - 在R中汇总组均值时如何创建有条件的新组，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/52655179/