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

我尝试过匹配和其他一些功能，但没有成功

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

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

3 个答案: