按组聚合多个列[重复]

Question

我有一个数据表如下：

library(data.table)
dt <- fread(
    "A   B   D  E  iso   year   
     1   A   1  NA ECU   2009   
     2   B   2  0  ECU   2009   
     3   D   3  0  BRA   2011   
     4   E   4  0  BRA   2011   
     5   D   7  NA ECU   2008   
     6   E   1  0  ECU   2008   
     7   A   3  2  BRA   2012   
     8   A   4  NA BRA   2012",
  header = TRUE
)

我想通过执行以下操作来创建数据集的聚合：dt[, .(D = sum(D)), by = c("iso", "year")]

但是，我希望能够添加A、E（以及实际数据中的另外 10 列），而不是仅用于 D。

期望的输出：

dt <- fread(
    "A   D  E  iso   year   
     3   3  0  ECU   2009   
     7   7  0  BRA   2011    
    11   8  0  ECU   2008    
    15   7  2  BRA   2012",
  header = TRUE
)

如何指定？

Answer 1

您可以将.SD + .SDcols 组合起来，然后循环.SD 和lapply()：

dt[, 
   lapply(.SD, sum, na.rm = TRUE), 
   by = c("iso", "year"), 
   .SDcols = c("A", "D", "E")]

   iso year  A D E
1: ECU 2009  3 3 0
2: BRA 2011  7 7 0
3: ECU 2008 11 8 0
4: BRA 2012 15 7 2

Answer 2

使用 dplyr 包

您也可以使用dplyr 包，在这种情况下您不需要指定所有列名：

library(dplyr)
dt$E[is.na(dt$E)] <- 0
dt %>%
  group_by(iso, year) %>%
  summarize_at(vars(A, D:E), .funs = sum)

输出：

# A tibble: 4 x 5
# Groups:   iso [2]
  iso    year     A     D     E
  <chr> <int> <int> <int> <dbl>
1 BRA    2011     7     7     0
2 BRA    2012    15     7     2
3 ECU    2008    11     8     0
4 ECU    2009     3     3     0

希望这会有所帮助。