如何在R的for循环中嵌入if语句修改dataframe的C-section列值

原始数据框 #

需求说明 #

若操作1至操作3中任意列的值为'A'、'B'或'F'，则将该行的C-section列值改为1，期望结果如下：

期望结果数据框 #

错误代码分析 #

你尝试的循环代码未生效，主要有以下问题：

for(i in 1:3){
    if (df[ , i] == 'A'){
      df$csection[, i] <- 1
  } 
    else if(df[ , i] == 'B'){
      df$csection[, i] <- 1
  } 
    else if(df[ , i] == 'F'){
      df$csection[, i] <- 1
}
}

• 列名不匹配：原始列名是C-section，代码中写的df$csection，R对列名大小写敏感，导致修改的是不存在的列。
• 整列判断逻辑错误：if(df[, i] == 'A')会对整列做比较，返回布尔向量，但if只能处理单个逻辑值，只会用第一行的判断结果，无法覆盖所有行。
• 索引错误：df$csection[, i]是错误写法，df$csection是一维向量，不能使用二维索引，无法定位到具体行。

正确解决方案 #

方法1：向量化操作（R推荐的高效写法） #

无需循环，直接按行判断并赋值：

# 按行检查操作1-3列是否包含目标值，转成0/1整数
df$`C-section` <- as.integer(apply(df[, 1:3], 1, function(row) {
  any(row %in% c('A', 'B', 'F'), na.rm = TRUE)
}))

• apply(df[,1:3], 1, ...)：对操作1-3列按行处理
• any(row %in% c('A','B','F'), na.rm=TRUE)：判断当前行是否有元素属于目标集合，na.rm=TRUE忽略NA的干扰
• as.integer()：将布尔值（TRUE/FALSE）转换为1/0，符合需求格式

方法2：修正循环写法（若偏好循环逻辑） #

如果一定要用循环，需遍历每行而非每列：

# 初始化C-section为0（原始数据已为0，可省略）
df$`C-section` <- 0

# 遍历每一行
for (row_idx in 1:nrow(df)) {
  # 检查当前行的操作1-3列是否包含目标值
  if (any(df[row_idx, 1:3] %in% c('A', 'B', 'F'), na.rm = TRUE)) {
    df$`C-section`[row_idx] <- 1
  }
}

• 遍历每行索引，针对单个行做判断
• 正确使用df$C-section[row_idx]修改对应行的值

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