标题说明它。R中的布尔向量之间的索引序列
我矢量
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
标题说明它。R中的布尔向量之间的索引序列
我矢量
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
#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
你可以使用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()
确实胁迫我们。
您还可以使用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
我们可以使用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
通过使用碱基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
'TF == FALSE'更简单'!TF'。另外,首先分配'cumsum(!TF)'会更有效率,因为你不需要计算两次。 –
@RichScriven是的,这是伟大的建议,谢谢! :) – Wen
'序列(长度)* TF'! –
每个人都给出了很好的答案,我很欣赏,并会一一浏览,但这是最容易理解的。 –