在达到特定值时重置 cumsum

Question

我要重置 cumsum在一个向量上，当它达到某个值时。

例如。对于以下向量:

v <- c(3, 5, 2, 5, 3, 4, 5, 3, 1, 4)

预期输出为:

c(0, 0, 10, 0, 0, 22, 0, 30, 0, 0)

与 reset <- 10我可以将任务减少到标记完整整数后的第一个值:

res <- cumsum(v)
resd <- res/reset
resd
# [1] 0.3 0.8 1.0 1.5 1.8 2.2 2.7 3.0 3.1 3.5

预期输出是这样的:

c(F, F, T, F, F, T, F, T, F, F) # or 
c(0, 0, 1.0, 0, 0, 2.2, 0, 3.0, 0, 0)

我需要一种快速的方法来计算其中一个。

Answer 1

我的(改进的)解决方案:

v <- c(3, 5, 2, 5, 3, 4, 5, 3, 1, 4)
res <- cumsum(v)
reset <- 10
resd <- res/reset
res[diff(c(0, floor(resd))) == 0] <- 0
print(res) #gives 0  0 10  0  0 22  0 30  0  0

编辑:现在是 v 中的第一个元素可以大于 10。