我有一个简单的二进制向量a
,我试图使用R函数cumsum
将其转换为向量b
。然而,cumsum
并不完全返回向量b
。
这里有一个例子:
a <- c(1,0,0,0,1,1,1,1,0,0,1,0,0,0,1,1)
b <- c(1,2,2,2,3,4,5,6,7,7,8,9,9,9,10,11)
> cumsum(a)
[1] 1 1 1 1 2 3 4 5 5 5 6 6 6 6 7 8
问题是,每当0出现在向量a
中时,前一个数字应该增加1,但仅限于前一个0。其余的值相同。
任何建议都很好!:-)
诀窍是使用diff
来标记转换:
cumsum(as.logical(a+c(0,abs(diff(a)))))
[1] 1 2 2 2 3 4 5 6 7 7 8 9 9 9 10 11