R语言含NA值时按行取最大值新增Winner列代码异常求助

代码保留NA值时运行失效，核心是两个R语言基础逻辑没有处理正确：

• 所有和NA做的比较运算（包括==、>、<等）返回结果永远是NA，而R的if判断仅识别明确的TRUE值，碰到NA会直接跳过当前分支，不会执行内部赋值。举个最常见的场景：如果某行Can1是NA，哪怕你用max(na.rm=TRUE)算出了正确的最高得票率，第一个判断最高值 == NA会返回NA，if分支直接失效，后续分支也无法正常匹配。
• 你在每个if分支里重复调用max()计算最大值，效率低且容错性差，只要单次计算受NA影响出问题，整行的Winner赋值就会失败。

方案1：修复原有逐行循环逻辑 #

如果要保留你原来的逐行遍历写法，只需要把得票率合并为向量统一计算最大值，用位置索引替代逐值比较即可，修复后的核心代码：

# 先初始化Winner列为字符型空值
df$Winner <- NA_character_

# 外层逐行循环
for (row in 1:nrow(df)) {
  # 提取当前行5个候选人的得票率
  vote_pct <- c(
    df[row, "Candidate1_percent"],
    df[row, "Candidate2_percent"],
    df[row, "Candidate3_percent"],
    df[row, "Candidate4_percent"],
    df[row, "Candidate5_percent"]
  )
  # 当前行所有得票率都是NA时直接跳过
  if (all(is.na(vote_pct))) next
  # 统一计算非NA最大值
  max_pct <- max(vote_pct, na.rm = TRUE)
  # 匹配最大值对应的候选人位置
  winner_pos <- which(vote_pct == max_pct)
  # 并列最高时默认取第一个候选人，可按需修改为标记"并列"等逻辑
  df[row, "Winner"] <- df[row, paste0("Candidate", winner_pos[1], "_name")]
}

方案2：向量化写法（推荐，无需显式循环） #

R原生支持逐行向量化计算，不用写冗长的for循环，代码更简洁、运行速度更快（数据量越大优势越明显），base R实现代码如下：

# 定义得票率列、候选人姓名列的列名规则
pct_cols <- paste0("Candidate", 1:5, "_percent")
name_cols <- paste0("Candidate", 1:5, "_name")

# 逐行计算Winner
df$Winner <- apply(df, 1, function(current_row) {
  pct_vals <- as.numeric(current_row[pct_cols])
  # 全NA行返回空值
  if (all(is.na(pct_vals))) return(NA_character_)
  # which.max自动忽略NA值，返回第一个最大值的位置
  winner_pos <- which.max(pct_vals)
  return(current_row[name_cols[winner_pos]])
})

小提示：which.max()本身自带忽略NA的逻辑，从根源上避免了手动做==比较时碰到NA返回空值的问题，鲁棒性比手动写max加判断更强。如果需要处理并列最高的场景，可以在找到winner_pos后判断长度，长度大于1时赋值为"得票并列"即可。

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