计算条件累计和

Question

我有以下数据帧：

N=seq(1:18)
Pre=c(0.5, 0.4, 0.7, 0.9, 0.6, 0.3, 0.7, 0.8, 0.6, 0.5, 0.4, 0.7, 0.9, 0.6, 0.3, 0.7, 0.8, 0.6)
df1=data.frame(N, Pre)

E=c(3.1, 2.9, 4.4)
Category=c(22,35,36)
df2=data.frame(E, Category)

我想在df1中创建两个新列(cumsum和category)

cumsum列中的值将不断累积，直到该值等于或小于df2$E的值，当该值大于该值时，cumsum将重新开始。在category列中，该值将是22、35或36，依此类推。

结果将是

​

​

这是一个简短的例子，我需要用一个非常大的数据帧来做这件事。

Answer 1

要获得Category列，我们可以cut cumsum的Pre值，并将df2的breaks定义为cumsum of E，将labels定义为相应的Category值。现在，我们可以使用这个Category值作为分组变量，并使用cumsum获取每个组中Pre值的ave。

df1$Category <- cut(cumsum(df1$Pre), breaks = c(0, cumsum(df2$E)), 
                                    labels = df2$Category)

df1$Cumsum <- ave(df1$Pre, df1$Category, FUN = cumsum)


df1
#    N Pre Category Cumsum
#1   1 0.5       22    0.5
#2   2 0.4       22    0.9
#3   3 0.7       22    1.6
#4   4 0.9       22    2.5
#5   5 0.6       22    3.1
#6   6 0.3       35    0.3
#7   7 0.7       35    1.0
#8   8 0.8       35    1.8
#9   9 0.6       35    2.4
#10 10 0.5       35    2.9
#11 11 0.4       36    0.4
#12 12 0.7       36    1.1
#13 13 0.9       36    2.0
#14 14 0.6       36    2.6
#15 15 0.3       36    2.9
#16 16 0.7       36    3.6
#17 17 0.8       36    4.4
#18 18 0.6     <NA>    0.6