返回基于布尔值的数字序列的更优雅的方法？

tcash21 发表于 Dev

tcash21

这是我作为data.frame一部分的布尔值示例：

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

我想从每个FALSE返回一个从1开始并递增1直到下一个FALSE的数字序列。

所得的期望向量为：

[1]  1  2  3  4  5  6  7  8  9 10  1  2  3  4  5  6  7  8  9 10  1

这是完成此任务的代码，但是我敢肯定，有一种更简单或更优雅的方法可以在R中执行此操作。我一直在尝试学习如何在R中更有效地编写代码，而不是简单地完成工作。

result <- c()
x <- 1
for(i in 1:length(atest)){
    if(atest[i] == FALSE){
        result[i] <- 1
        x <- 1
    } 
    if(atest[i] != FALSE){
        x <- x+1
         result[i] <- x
    }
}

约书亚·乌尔里希（Joshua Ulrich）

这是使用方便（但不广为人知/使用）的基本功能的一种方法：

> sequence(tabulate(cumsum(!atest)))
 [1]  1  2  3  4  5  6  7  8  9 10  1  2  3  4  5  6  7  8  9 10  1

分解：

> # return/repeat integer for each FALSE
> cumsum(!atest)
 [1] 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3
> # count the number of occurrences of each integer
> tabulate(cumsum(!atest))
[1] 10 10  1
> # create concatenated seq_len for each integer
> sequence(tabulate(cumsum(!atest)))
 [1]  1  2  3  4  5  6  7  8  9 10  1  2  3  4  5  6  7  8  9 10  1

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。