r - 使用 sum、length 和 grep 聚合 data.table

Question

让我们制作一个 data.table:

dt <- data.table(x.1=1:8, x.2=1:8, x.3=2:9, vessel=rep(letters[1:2], each=4), Year=rep(2012:2015, 2))
dt
   x.1 x.2 x.3 vessel Year
1:   1   1   2      a 2012
2:   2   2   3      a 2013
3:   3   3   4      a 2014
4:   4   4   5      a 2015
5:   5   5   6      b 2012
6:   6   6   7      b 2013
7:   7   7   8      b 2014
8:   8   8   9      b 2015

我可以使用函数 length 和 sum 对它进行聚合，以获得每年所有 x 的总和以及每年独特船只的总和，如下所示:

dt[, 
            list(
  x.1=sum(x.1),
  x.2=sum(x.2),
  x.3=sum(x.3),
  vessels=length(unique(vessel))),
    by=list(Year=Year)]

   Year x.1 x.2 x.3 vessels
1: 2012   6   6   8       2
2: 2013   8   8  10       2
3: 2014  10  10  12       2
4: 2015  12  12  14       2

这就是我想要的，但在我的真实数据中我有很多列，所以我想使用 grep 或 %like%，但我无法让它工作。我在想一些与此相符的事情:

dt[,grep("x",colnames(dt)),with = FALSE])

但是如何将其与聚合合并？

Answer 1

您可以使用 lapply对所有( .SD )或几列(使用 .SDcols 选择)应用函数:

dt[, lapply(.SD, sum), by=Year, .SDcols=c("x.1","x.2")]

以下内容也可能适用于选择名称中带有“x”的所有列:

dt[, c(lapply(.SD, sum), vessel=uniqueN(vessel)),
    by=Year,
    .SDcols=grepl("^x", names(dt))
]