r - R - 按条件累积总和

Question

所以我有一个简化的数据集，看起来像这样：

Year    ID     Sum
2009    999    100
2009    123     85
2009    666    100
2009    999    100
2009    123     90
2009    666     85
2010    999    100
2010    123    100
2010    666     95
2010    999     75
2010    123    100
2010    666     85

我想按年份和 ID 添加一个包含累计总和的列。像这样：

Year     ID     Sum    Cum.Sum
2009    999     100        100
2009    123      85         85  
2009    666     100        100  
2009    999     100        200
2009    123      90        175
2009    666      85        185
2010    999     100        100
2010    123     100        100
2010    666      95         95
2010    999      75        175
2010    123     100        200
2010    666      85        180

我认为这应该很简单，但不知何故我无法弄清楚。我该怎么做呢？谢谢您的帮助！

Answer 1

使用data.table：

require(data.table)
DT <- data.table(DF)
DT[, Cum.Sum := cumsum(Sum), by=list(Year, ID)]

    Year  ID Sum Cum.Sum
 1: 2009 999 100     100
 2: 2009 123  85      85
 3: 2009 666 100     100
 4: 2009 999 100     200
 5: 2009 123  90     175
 6: 2009 666  85     185
 7: 2010 999 100     100
 8: 2010 123 100     100
 9: 2010 666  95      95
10: 2010 999  75     175
11: 2010 123 100     200
12: 2010 666  85     180

Answer 2

其他方式

1）使用ddply按组对变量求和（类似于SQL group by）

X <- ddply ( dataset, .(Year,ID), sum)

2）将结果与数据集合并

Y <- merge( dataset, X, by=('Year','ID')

Answer 3

您可以使用 dplyr 和基本函数cumsum：

require(dplyr)    

dataset %>% 
  group_by(Year, ID) %>%
  mutate(cumsum = cumsum(Sum)) %>%
  ungroup()