r - 在 dplyr 中构建动态行数的总和

Question

我的 df 看起来像下面的前三列:

ID VAL LENGTH SUM
1  1   1      1
1  1   1      1
1  1   2      2
1  1   2      2
2  0   1      0
2  3   1      0
2  4   2      3

我想添加第四列，它被定义为 VAL 中组的第一个到 LENGTH-st 值的总和。

我该怎么做？

Answer 1

你可以这样做:

library(dplyr)

df %>%
  group_by(ID) %>%
  mutate(SUM = sapply(LENGTH, function(x) sum(VAL[1:x])))

输出:

# A tibble: 7 x 4
# Groups:   ID [2]
     ID   VAL LENGTH   SUM
  <int> <int>  <int> <dbl>
1     1     1      1     1
2     1     1      1     1
3     1     1      2     2
4     1     1      2     2
5     2     0      1     0
6     2     3      1     0
7     2     4      2     3