如何在R数据帧的每个方向上获得满足给定条件的每行周围的k行？

# 示例数据帧
df <- data.frame(
  x = c(1, 2, 3, 4, 5),
  y = c(6, 7, 8, 9, 10),
  z = c(11, 12, 13, 14, 15)
)

# 给定条件
condition <- df$x > 2

# 每个方向上的k值
k <- 1

# 在行方向上获取满足条件的每行周围的k行
rows <- which(condition)
result_rows <- lapply(rows, function(row) {
  start <- max(1, row - k)
  end <- min(nrow(df), row + k)
  df[start:end, ]
})

# 在列方向上获取满足条件的每行周围的k行
cols <- which(condition, arr.ind = TRUE)
result_cols <- lapply(cols, function(col) {
  start <- max(1, col - k)
  end <- min(ncol(df), col + k)
  df[, start:end]
})

# 输出结果
print(result_rows)
print(result_cols)