Cumulative sum of a column based on the values ​​of another column (R)

I have a data frame with 2 columns, like this:

> data.frame(x=1:10, y=c(0,0,0,1,1,0,0,1,0,1))
    x y
1   1 0
2   2 0
3   3 0
4   4 1
5   5 1
6   6 0
7   7 0
8   8 1
9   9 0
10 10 1

and I want to get the cumulative sum of the column x( cumsum(df$x)), but the sum should be reset after appearing 1in the column y. This is the result I'm looking for:

1
3
6
10
5
6
13
21
9
19

How can I achieve this in R?

+3
source share
2 answers

You can achieve this using ave:

ave(d$x,c(0,cumsum(d$y[-nrow(d)])),FUN=cumsum)

#  [1]  1  3  6 10  5  6 13 21  9 19
+3
source

A data.tableusingshift

 library(data.table) #devel version `data.table_1.9.5` 
 setDT(d)[, cumsum(x), by = cumsum(shift(y, fill=0))]$V1
 #[1]  1  3  6 10  5  6 13 21  9 19
+3
source

Source: https://habr.com/ru/post/1685757/


All Articles