r - 仅正数的累积和

Question

这个问题在这里已经有了答案:





Create counter within consecutive runs of certain values

(6 个回答)


6 个月前关闭。




我有这个向量:

x = c(1,1,1,1,1,0,1,0,0,0,1,1)

而且我只想对正数进行累积总和。我应该得到以下向量作为返回:

xc = (1,2,3,4,5,0,1,0,0,0,1,2)

我怎么能做到？

我试过: cumsum(x)但是对所有值进行累积总和并给出:

cumsum(x)
[1] 1 2 3 4 5 5 6 6 6 6 7 8

Answer 1

一种选择是

x1 <- inverse.rle(within.list(rle(x), values[!!values] <- 
                  (cumsum(values))[!!values]))
x[x1!=0] <- ave(x[x1!=0], x1[x1!=0], FUN=seq_along)
x
#[1] 1 2 3 4 5 0 1 0 0 0 1 2

或者一行代码是

 x[x>0] <-  with(rle(x), sequence(lengths[!!values]))
 x
 #[1] 1 2 3 4 5 0 1 0 0 0 1 2