问如何在预定义的值之前和之后选择特定数量的行

EN

我正在尝试从大量的时间序列数据集中选择相关的行。棘手的一点是，所需的行在列中的某些值之前和之后。

# example data
x <- rnorm(100)
y <- rep(0,100)
y[c(13,44,80)] <- 1
y[c(20,34,92)] <- 2
df <- data.frame(x,y)

在本例中，df$y列中的临界值为1和2。例如，如果我想选择df$y==1之前的2行和之后的4行，我可以这样做：

ones<-which(df$y==1)
selection <- NULL
for (i in ones) {
  jj <- (i-2):(i+4)
  selection <- c(selection,jj)
}
df$selection <- 0
df$selection[selection] <- 1

可以说，这对于更多的值来说，可伸缩性很差。对于df$y==2，我必须重复以下内容：

twos<-which(df$y==2)
selection <- NULL
for (i in twos) {
  jj <- (i-2):(i+4)
  selection <- c(selection,jj)
}
df$selection[selection] <- 2

理想的场景是这样一个函数，它执行类似于这个虚构函数selector(data=df$y, values=c(1,2), before=2, after=5, afterafter = FALSE, beforebefore=FALSE)的操作，其中向values提供临界值，向before提供之前要选择的行数，并相应地向after提供行数。

然而，afterafter允许从某些行到值之后的某些行，例如after=5,afterafter=10 (与afterafter相同，但进入另一个方向)。

欢迎提出任何建议和建议！谢谢!