R语言技术问询：如何基于标识列替换数据框指定NA为0

基础R实现代码 #

基于你已构造的数据框，结合标识列D完成替换：

# 初始化数据框
A<-c(NA,NA,1,1)
B<-c(NA,NA,NA,1)
C<-c(1,NA,NA,NA)
df<-data.frame(A,B,C)
df$D<-ifelse(!is.na(df$A) | !is.na(df$B) | !is.na(df$C), 1,0)

# 替换NA为0（仅D=1的行）
df[, c("A", "B", "C")] <- lapply(df[, c("A", "B", "C")], function(x) {
  ifelse(df$D == 1, replace(x, is.na(x), 0), x)
})

# 查看结果
df

执行后输出：

A B C D
1 0 0 1 1
2 NA NA NA 0
3 1 0 0 1
4 1 1 0 1

tidyverse（dplyr）实现代码 #

若习惯用tidyverse语法，可通过across批量处理列：

library(dplyr)

# 初始化数据框（同上）
A<-c(NA,NA,1,1)
B<-c(NA,NA,NA,1)
C<-c(1,NA,NA,NA)
df<-data.frame(A,B,C)
df$D<-ifelse(!is.na(df$A) | !is.na(df$B) | !is.na(df$C), 1,0)

# 替换逻辑
df <- df %>%
  mutate(across(A:C, ~if_else(D == 1, replace_na(.x, 0), .x)))

逻辑说明 #

• 当标识列D为1时，说明该行存在有效数据，将其中的NA替换为0
• 当标识列D为0时，整行全为NA，保留原始缺失值

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