向量索引的出现数

Question

我目前正在优化R代码。所以，我尽量避免动态内存分配，循环，.但是，对于FOR循环，我有一些困难，我需要创建这样的行为：

投入：

v <- c(TRUE,FALSE,FALSE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,FALSE)

产出：

 [1] 0 0 0 1 1 2 3 3 3 3

我认为，最简单的方法是能够计算向量当前索引中出现的"TRUE“数。如果你有任何想法..。记住要避免FOR循环和非优化代码.

谢谢

Answer 1

这就是诀窍：

cumsum(v)-1L