R：没有for循环的data.table中的列总和答案

【问题标题】：R: column sum in a data.table without for-loopR：没有for循环的data.table中的列总和
【发布时间】：2015-02-09 18:27:50
【问题描述】：

我有一个如下结构的data.table：

DT = data.table(ID = rep(c("A", "B"), each=5), 
                DY = 1:5, 
                S = c(0, 50, 50, 200, 50, 5, 20, 5, 20, 25))

此data.table 应添加一列C，其条目只是按ID 分组的1:5 中DY i 的总和。上面例子的结果是

DT[, C:=c(0, 50, 100, 300, 350, 5, 25, 30, 50, 75)]

正如我所描述的，for 循环是获得我想要的结果的第一选择。但是，有没有更好的方法在没有 for 循环的情况下计算 C，尤其是使用 data.table-package 的强大功能？

【问题讨论】：

【解决方案1】：

根据显示的结果，您可能需要按“ID”分组的“S”列的cumsum

 DT[, C:=cumsum(S), by = ID]

如果您需要按照@eddi 的建议进行改造

 DT[, S:=c(C[1L],diff(C)), by = ID]

【讨论】：