每当使用 R 在向量中为零时重置累积和

Question

只要使用 R.E.g. 向量中有零，我就需要重置累积总和输入 c(0,1,0,0,0,1,1,1,0,1,1) 我希望输出为 0,1,0,0,0,1,2,3,0,1, 2.我检查了多个回答使用函数重置序列的答案，但这些解决方案在这里不起作用。

Numeric sequence with condition , Create counter with multiple variables , Comp. Efficent way of resetting sequence if condition met ( R )是我提到的一些。我尝试了 cumsum、ave 和 getanID 的不同组合，但似乎无法获得我想要的输出。

Answer 1

也许是这样的:

vec <- c(0,1,0,0,0,1,1,1,0,1,1)
library(data.table)
as.numeric(unlist(by(vec, rleid(vec), cumsum))) #or as in Maurits Evers answer `unname`, or `unlist(..., use.names = F)` instead of `as.numeric`
#output
 0 1 0 0 0 1 2 3 0 1 2

rleid 生成游程类型 id 列:

rleid(vec)
#output
1 2 3 3 3 4 4 4 5 6 6

然后将其用作分组变量

编辑:根据@digEmAll 的建议:

请注意，这仅在 vec 仅包含 0 和 1 时有效。要使其更通用，您应该使用

rleid(vec > 0)