R Выбор строки и следующих 5 строк

Question

В моей кадре данных я хотел бы выбрать строку на основе некоторой логики, а затем вернуть кадр данных с выбранной строкой PLUS в следующие строки «N».

Итак, у меня есть это: (общий пример)

workingRows <- myData[which(myData$Column1 >= myData$Column2 & myData$Column3 <= myData$Column4), ] 

Который возвращает меня правильные "стартовые значения". Как я могу получить «следующие» 5 значений на основе каждого из стартовых значений?

Answer 1

Мы можем использовать rep, чтобы получить следующие 5 строк, sort это, и если есть какие-либо дубликаты от наложений, обертывают его с unique и подмножество в «MyData».

i1 <- which(myData$Column1 >= myData$Column2 & myData$Column3 <= myData$Column4) 
myData[unique(sort(i1 + rep(0:5, each = length(i1)))),]