如何在R中将每行首次出现1之后的所有值设为NA

基础R方案 #

直接使用apply()逐行处理，自定义函数定位每行第一个1的位置，将后续元素替换为NA：

# 构建原始数据
T1 <- c(0,0,0,0,0)
T2 <- c(1,0,0,0,0)
T3 <- c(0,1,0,0,0)
T4 <- c(1,1,0,NA,1)
T5 <- c(0,1,0,NA,0)
df <- data.frame(T1,T2,T3,T4,T5)

# 定义行处理函数
process_row <- function(row) {
  # 定位第一个1的位置（自动忽略NA）
  first_one <- which(row == 1)[1]
  # 若存在1，将后续元素设为NA
  if (!is.na(first_one)) {
    row[(first_one + 1):length(row)] <- NA
  }
  return(row)
}

# 应用到所有行并转换回数据框
df_processed <- as.data.frame(t(apply(df, 1, process_row)))
colnames(df_processed) <- colnames(df)

# 查看结果
df_processed

Tidyverse方案（dplyr + purrr） #

如果习惯使用tidyverse工具链，可结合pmap_dfr()实现行级操作：

library(dplyr)
library(purrr)

df_processed <- df %>%
  pmap_dfr(function(...) {
    row <- c(...)
    first_one <- which(row == 1)[1]
    if (!is.na(first_one)) {
      row[(first_one + 1):length(row)] <- NA
    }
    as.data.frame(t(row))
  })

colnames(df_processed) <- colnames(df)

说明 #

• 函数会自动忽略行中的NA值，仅识别第一个明确的1
• 对于没有出现1的行（如示例第3行），保持原数据不变
• 处理后的结果与需求完全匹配

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