含NA数据的嵌套ifelse()语句失效，请求排查问题原因

• 问题根源：当weight为NA时，第三个ifelse的条件data$weight == 1会返回NA，ifelse函数遇到条件为NA时，不会执行后续分支，直接返回NA，导致无法触发最后一个判断batch=1且weight=NA的逻辑。

• 解决方案1：调整条件顺序，将判断batch=1且weight=NA的分支提前，放在判断weight==1的分支之前：

data <- data.frame(batch = c(1, 1, 1, 1, 1, 1, 0, 0, NA),
          weight = c(1, 0, 0, 1, NA, 1, 1, NA, NA))

data$class <- ifelse((data$batch == 1 & data$weight == 0), 1,
                  ifelse((data$batch == 1 & data$weight == 1), 2,
                         ifelse((data$batch == 1 & is.na(data$weight)), 3,
                                ifelse((data$weight == 1), 4, 0))))

• 解决方案2：在判断weight==1时，先排除NA的情况，确保条件返回TRUE/FALSE而非NA：

data$class <- ifelse((data$batch == 1 & data$weight == 0), 1,
                  ifelse((data$batch == 1 & data$weight == 1), 2,
                         ifelse((data$weight == 1 & !is.na(data$weight)), 4, 
                                ifelse((data$batch == 1 & is.na(data$weight)), 3, 0))))

测试修正后的代码，第5行（batch=1，weight=NA）的class会被正确设为3。

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