如何在R语言数据框中找到首次出现的1并将后续值替换为NA

我有一个包含二进制值（0,1）的R语言数据框df，其结构如下：

'data.frame':   10 obs. of  7 variables:
 $ Round.1: int  0 0 0 1 1 0 0 0 1 0
 $ Round.2: int  0 0 0 0 1 0 0 0 0 0
 $ Round.3: int  0 0 0 0 1 0 0 0 1 0
 $ Round.4: int  0 0 0 1 0 0 0 0 0 0
 $ Round.5: int  0 0 0 1 1 0 0 0 0 0
 $ Round.6: int  0 0 0 0 0 0 1 0 1 0
 $ Round.7: int  0 0 0 0 1 0 1 0 0 0

需求 #

需要完成两个操作：

• 找到每列中第一个1的出现位置
• 将该位置之后的所有值替换为NA

期望结果 #

处理后的数据框结构如下：

'data.frame':   10 obs. of  7 variables:
 $ Round.1: int  0 0 0 1 1 0 0 0 1 0
 $ Round.2: int  0 0 0 NA NA 0 0 0 NA 0
 $ Round.3: int  0 0 0 NA NA 0 0 0 NA 0
 $ Round.4: int  0 0 0 NA NA 0 0 0 NA 0
 $ Round.5: int  0 0 0 NA NA 0 0 0 NA 0
 $ Round.6: int  0 0 0 NA NA 0 1 0 NA 0
 $ Round.7: int  0 0 0 NA NA 0 NA 0 NA 0

解决方案 #

可以通过自定义函数结合lapply实现逐列处理：

# 定义处理单个列的函数
process_single_col <- function(col) {
  # 获取第一个1的索引，无1则返回原列
  first_1_pos <- which(col == 1)[1]
  if (!is.na(first_1_pos)) {
    # 将第一个1之后的所有值替换为NA
    col[(first_1_pos + 1):length(col)] <- NA
  }
  return(col)
}

# 应用函数到数据框的每一列
df_processed <- as.data.frame(lapply(df, process_single_col))

代码说明 #

• which(col == 1)[1]：定位当前列中第一个值为1的位置，若列中没有1则返回NA
• 当存在第一个1时，从该位置的下一位开始到列末尾，全部替换为NA
• lapply遍历数据框的每一列执行处理逻辑，最后用as.data.frame将结果转回数据框格式

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