如何将R数据框中1之后的连续数值置为0？

基础R实现 #

首先构造你的数据框：

df <- data.frame(a = c(0,0,0,0,1,1,1,0,0,0),
                 b = c(0,0,1,1,1,1,0,0,0,0),
                 c = c(1,1,0,0,0,0,0,1,1,0))

定义处理单列的函数，找到第一个1的位置后，将其后续所有值置为0：

process_col <- function(col) {
  first_one <- which(col == 1)[1]
  if (!is.na(first_one)) {
    col[(first_one + 1):length(col)] <- 0
  }
  col
}

# 应用到所有列并转换回数据框
df_processed <- as.data.frame(apply(df, 2, process_col))

查看处理结果：

print(df_processed)
#>    a b c
#> 1  0 0 1
#> 2  0 0 0
#> 3  0 0 0
#> 4  0 0 0
#> 5  1 0 0
#> 6  0 0 0
#> 7  0 0 0
#> 8  0 0 0
#> 9  0 0 0
#> 10 0 0 0

dplyr风格实现 #

如果你习惯使用tidyverse工具，可以用dplyr::across()实现：

library(dplyr)

df_processed <- df %>%
  mutate(across(everything(), ~{
    first_one <- which(.x == 1)[1]
    if (!is.na(first_one)) {
      .x[(first_one + 1):length(.x)] <- 0
    }
    .x
  }))

两种方法都能实现需求：保留每列第一个出现的1，将其之后的所有数值（包括后续的1）置为0；如果列中没有1，则保持原数据不变。

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