Нахождение начального и конечного индексов последовательно в R

Предположим, у меня есть последовательность:

x = c( 1, 1, 0, 0, 1, 0, 1, 1, 1, 0, 0)

Есть ли в R элегантный способ вернуть начальный и конечный индексы каждой последовательности 1?

Ответ должен быть массивом из 2 столбцов с nRows = количество последовательностей 1:

startIndx = [ 1, 5, 7 ]
stopIndex = [ 2, 5, 9 ]

Спасибо.

БСЛ


r indices sequence pattern-matching
person BLevy    schedule 21.03.2015    source источник

Ответы (5)


arrow_upward
8
arrow_downward

Предполагая, что ваш вектор состоит из значений 0 и 1:

which(diff(c(0L, x)) == 1L)
#[1] 1 5 7
which(diff(c(x, 0L)) == -1L)
#[1] 2 5 9

В противном случае вам сначала понадобится что-то вроде x <- x == 1L.

person Roland    schedule 21.03.2015

arrow_upward
5
arrow_downward

Элегантный способ

y <- which(x==1)
startIndx <- y[!(y-1) %in% y]
stopIndex <- y[!(y+1) %in% y]
rbind(startIndx, stopIndex)
#          [,1] [,2] [,3]
#startIndx    1    5    7
#stopIndex    2    5    9
person Khashaa    schedule 21.03.2015
comment
Смотрел на что ;) Отлично! - person Colonel Beauvel; 21.03.2015

arrow_upward
2
arrow_downward

Попробуй это:

y = rle(x)

stopIndex  = with(y, cumsum(lengths)[values==1])
startIndex = stopIndex - with(y, lengths[values==1]) + 1

#> stopIndex
#[1] 2 5 9
#> startIndex
#[1] 1 5 7
person Colonel Beauvel    schedule 21.03.2015

arrow_upward
2
arrow_downward

Что насчет этого? [отредактированная версия по предложению alexis_laz]

library(cgwtools)
res <- seqle(which(as.logical(x)))
rbind(res$values, res$values + res$lengths - 1)
     [,1] [,2] [,3]
[1,]    1    5    7
[2,]    2    5    9
person DatamineR    schedule 21.03.2015
comment
Если я что-то не упустил, вы можете заменить последнюю строку на rbind(res$values, res$values + res$lengths - 1) - person alexis_laz; 21.03.2015
comment
Спасибо за предложение, теперь намного легче читать/понимать :) - person DatamineR; 21.03.2015

arrow_upward
1
arrow_downward

Как насчет этого:

startIndx<-rev(length(x)-cumsum(rle(rev(x))$lengths)[rle(rev(x))$values==1]+1)
stopIndex<-cumsum(rle(x)$lengths)[rle(x)$values==1]
person Community    schedule 21.03.2015