基于最小5个连续值的向量序列的子集数据帧

Pat*_*Pat 6 r subset sequence dataframe

我有一个看起来像这样的矢量:

out1[1:200]
  [1] NA NA NA NA  0  1  2 NA NA NA  1 NA  0 NA  0  1 NA NA  0 NA  0  1  2  2  2 NA  0  1  2  3  4  4  5  6  7  8  9  9  9  9
 [41] 10 11 NA  0  0 NA  1 NA  0  1 NA  0 NA  0  1  2 NA  1 NA  0  0  0  1  2 NA NA NA  0  0 NA  0  0  0  1  2 NA  1  2 NA  0
 [81]  1  2  3  4  5  6  7  8 NA  0  1  2  3  4 NA  0  1  2  2  3  4  5 NA  0  1  2  3  3  4  5  5  6  7 NA  1  2 NA  1  2 NA
[121]  0  1  2 NA  1  2  3  3  3  3  4 NA  0  0  0  1  2  3  4  5 NA NA  0  1 NA NA NA  1  2  2  3 NA  1  2  2  2 NA NA  0  1
[161] NA  1 NA  1  2 NA  0  0 NA NA  0  1 NA NA NA NA  1  2  3 NA NA  1  2  3  4  5  6 NA  1  2  3  4  5  6  6  7  8 NA  0  1
Run Code Online (Sandbox Code Playgroud)

我现在想要df通过这个向量对a (具有相同的长度)进行子集化,但是只有具有超过最少5个连续数字的范围的序列,例如0:4或1:5(当然,所有这些都比这更长).因此,也NA应该如此FALSE.

例如

out1: NA NA 0 1 2 2 NA 0 0 1 2 3 3 4 NA 
Run Code Online (Sandbox Code Playgroud)

然后结果应该是

out2: F F F F F F F T T T T T T T F
Run Code Online (Sandbox Code Playgroud)

Kha*_*haa 5

以下给出了期望的结果

library(data.table) # v >= 1.9.5 (devel version - install from GitHub)
data.table(x)[,id:=rleid(!is.na(x)),
   ][ , aa:=(.N>5) , by = id
      ][ ,aaa:=4 %in% cumsum(diff(unique(sort(x)))), by = .(id, aa)
         ]$aaa

## [1] FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE  TRUE  TRUE  TRUE  TRUE  TRUE
## [15]  TRUE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE
Run Code Online (Sandbox Code Playgroud)

数据

x <- c(NA, NA, NA, NA, NA, 0, 1, 2, NA, 0, 1, 2, 3, 4, 4, 5, NA, 1, 2, 3, 3, 3, 3, 4, NA)
Run Code Online (Sandbox Code Playgroud)