r - 在R中查找长度大于1的向量元素

Question

我有一个像这样的向量

c(0,1,2,0,0,2,2,2,2,2,2,1,0,1,2,2,2,2,2,1)

我想找到一系列至少三个连续的 2，c(2,2,2) 开始的位置，如果它被打断，我想找到下一个第一个位置。

返回向量应如下所示：

FALSE FALSE FALSE FALSE FALSE TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE TRUE FALSE FALSE FALSE FALSE FALSE

我尝试了 match 和其他几个功能，但没有成功

Answer 1

这是一种方法：

x <- c(0,1,2,0,0,2,2,2,2,2,2,1,0,1,2,2,2,2,2,1)
a <- rle(x)
z <- rep(FALSE, length(x))
z[sequence(a$lengths) == 1] <- a$lengths >= 2 & a$values == 2
z
#  [1] FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE
# [11] FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE

Answer 2

r <- rle(x)
# Which entries are true in the result:

w <- (cumsum(r$length)[r$values==2] - (r$length[r$values==2]-1))[r$length[r$values==2]>2]
result <- logical(length(x))
result[w] <- T
result
##  [1] FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE
## [17] FALSE FALSE FALSE FALSE

Answer 3

这是我作为函数的尝试：

FUN <- function(x, n = 3) {
    y <- rle(x)
    z <- y[[1]] > (n - 1)
    unlist(lapply(1:length(z), function(i) {
        m <- rep(FALSE, each=y[[1]][i])
        if(z[i]) {
            m[1] <- TRUE
        }
        m
    }))
}

FUN(x)

##  [1] FALSE FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE
## [11] FALSE FALSE FALSE FALSE  TRUE FALSE FALSE FALSE FALSE FALSE