在R中为符合条件的行分配因子：新增条件判断列的实现

嘿，我来帮你搞定这个需求！要给数据框添加符合三种条件的因子列，我们有两种常用的方法，一种是用dplyr包的case_when（代码更清晰易读），另一种是基础R的嵌套ifelse，我都给你详细演示下。

首先先确认你的示例数据：

before <- data.frame( 
  a = c(1,3,2,2,4), 
  b = c(NA,2,3,NA,3), 
  c = c(NA,NA,NA,NA,5) 
)

方法一：使用dplyr的case_when（推荐） #

case_when可以让我们像写多分支条件一样清晰，而且支持按顺序匹配条件，注意一定要把最严格的条件放在前面（比如三个列都有值的情况要先判断，不然会被宽松的条件提前匹配）。

代码如下：

library(dplyr)

after <- before %>%
  mutate(level = case_when(
    # 条件3：a、b、c都有非NA值
    !is.na(a) & !is.na(b) & !is.na(c) ~ 3,
    # 条件2：a和b有值，c是NA
    !is.na(a) & !is.na(b) & is.na(c) ~ 2,
    # 条件1：仅a有值，b和c都是NA
    !is.na(a) & is.na(b) & is.na(c) ~ 1
  )) %>%
  # 将level列转换为因子类型
  mutate(level = as.factor(level))

# 查看结果
print(after)

运行后你会得到预期的结果：

a  b  c level
1 1 NA NA     1
2 3  2 NA     2
3 2  3 NA     2
4 2 NA NA     1
5 4  3  5     3

方法二：基础R的嵌套ifelse #

如果你不想加载额外的包，用基础R的ifelse嵌套也能实现，逻辑和上面一致，只是写法稍微繁琐一点：

before$level <- ifelse(
  !is.na(before$a) & !is.na(before$b) & !is.na(before$c), 
  3,
  ifelse(
    !is.na(before$a) & !is.na(before$b) & is.na(before$c), 
    2,
    ifelse(
      !is.na(before$a) & is.na(before$b) & is.na(before$c), 
      1,
      NA  # 如果有不符合上述三种的情况，赋值NA
    )
  )
)

# 转换为因子
before$level <- as.factor(before$level)

print(before)

这个方法的输出和上面完全一致，适合不想依赖第三方包的场景。

关键注意点 #

• 条件的顺序很重要：一定要从最严格的条件（比如三个列都有值）开始判断，否则宽松的条件会提前匹配，导致后面的条件无法触发。
• 如果你需要处理其他边缘情况（比如a列也是NA的情况），可以在最后添加对应的条件或者赋值NA。

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