如何在R语言DataFrame中将0值行及后续5行统一设为0？

我有一个R语言DataFrame，其中部分行的n列值为0，需要编写代码将每个0值行及其后续5行的n列值全部设为0。

当前数据示例 #

> head(df$n,n=20)
              df$n
1                0
2             9009
3                0
4                0
5                0
6                0
7                0
8             5410
9                0
10               0
11               0
12               0
13               0
14               0
15              32
16               0
17               0
18            1054
19               0
20               0

期望输出 #

> head(df$n,n=20)
                  df$n
    1                0
    2                0
    3                0
    4                0
    5                0
    6                0
    7                0
    8             5410
    9                0
    10               0
    11               0
    12               0
    13               0
    14               0
    15              32
    16               0
    17               0
    18               0
    19               0
    20               0

尝试过的代码及问题 #

• 嵌套for循环，实现效果不符合预期：

for(j in 1:nrow(indx)){
for(i in 1:4){
  df$n[j+i]<-0
}
}

问题：j是存储0值行号的DataFrame的行索引，不是原数据的行号，导致后续行定位错误。

• 手动赋值的循环，能实现需求但会在末尾生成额外行：

for( i in 1:nrow(indx)){
  u<-indx[i,]
  df[u,]<-0
  df[u+1,]<-0
  df[u+2,]<-0
  df[u+3,]<-0
  df[u+4,]<-0
  df[u+5,]<-0
}

问题：当u+5超过原数据总行数时，R会自动扩展数据框生成额外行。

正确实现方案 #

采用向量化操作标记目标行，同时处理边界避免生成额外行：

# 获取数据总行数
total_rows <- nrow(df)
# 找出n列为0的所有行索引
zero_rows <- which(df$n == 0)
# 生成所有需要置0的行号：每个0行及其后5行，且不超过总行数
target_rows <- unique(unlist(lapply(zero_rows, function(x) x:min(x+5, total_rows))))
# 将目标行的n列设为0
df$n[target_rows] <- 0

方案说明 #

1. 用which(df$n == 0)精准定位原数据中n列为0的行索引，避免依赖额外的indx数据框。
2. 通过min(x+5, total_rows)确保不会超出数据总行数，杜绝额外行生成。
3. 用unique()去重，避免重复标记同一行（比如连续的0值行可能导致重复覆盖）。
4. 向量化操作比循环更高效，逻辑也更清晰。

内容的提问来源于stack exchange，提问作者CovetTachi