R中布尔向量之间的顺序索引

Question

标题如是说。

我的矢量

TF <- c(F,T,T,T,F,F,T,T,F,T,F,T,T,T,T,T,T,T,F)

我想要的输出

[1] 0 1 2 3 0 0 1 2 0 1 0 1 2 3 4 5 6 7 0

Answer 1

您可以将 rle() 与 sequence() 一起使用。

replace(TF, TF, sequence(with(rle(TF), lengths[values])))
# [1] 0 1 2 3 0 0 1 2 0 1 0 1 2 3 4 5 6 7 0

replace() 为我们强制转换。

Answer 2

#with(rle(TF), sequence(lengths) * rep(values, lengths))
with(rle(TF), sequence(lengths) * TF) #Like Rich suggested in comments
# [1] 0 1 2 3 0 0 1 2 0 1 0 1 2 3 4 5 6 7 0

Answer 3

您还可以使用 inverse.seqle 来自 cgwtools:

library(cgwtools)
SEQ = inverse.seqle(rle(TF)) 
SEQ[!TF] = 0

或类似于@d.b的回答：

inverse.seqle(rle(TF))*TF

# [1] 0 1 2 3 0 0 1 2 0 1 0 1 2 3 4 5 6 7 0

Answer 4

我们可以使用rleid

library(data.table)
ave(TF, rleid(TF), FUN = seq_along)*TF
#[1] 0 1 2 3 0 0 1 2 0 1 0 1 2 3 4 5 6 7 0

Answer 5

通过使用基数 R

ave(cumsum(TF==FALSE), cumsum(TF==FALSE), FUN=seq_along)-1
 [1] 0 1 2 3 0 0 1 2 0 1 0 1 2 3 4 5 6 7 0

R中布尔向量之间的顺序索引

Sequentially index between a boolean vector in R

r

seq