R语言函数与循环问题：编码匹配后标签无法正确填充

问题背景 #

编写的CleanConditionPreg函数意图通过CodeConditionPreg表的编码映射，将Birth_IndicatorsConditions中以Complications开头的列的编码替换为对应标签，但执行后生成的新列全为空白值。

原始数据表：

CodeConditionPreg表 #

Birth_IndicatorsConditions表 #

期望输出：

问题原因 #

1. 列名大小写不匹配：CodeConditionPreg表的编码列是Code（首字母大写），但原代码中使用CodeConditionPreg$code（小写）进行匹配，导致match()无法找到对应值，返回全NA。
2. 函数无返回值：原函数最后未通过return()返回处理后的数据集，导致调用后无法获取正确结果。
3. 潜在数据类型不匹配：若Complications列与Code列数据类型不一致（如一个是字符型、一个是因子型），也会导致匹配失败。

修复后的代码 #

library(dplyr)

CleanConditionPreg <- function(Condition){
  # 提取Complications开头的列
  codes <- Condition %>% select(starts_with("Complications"))
  
  # 确保CodeConditionPreg的Code列是字符型（避免类型不匹配）
  CodeConditionPreg <- CodeConditionPreg %>% mutate(Code = as.character(Code))
  
  # 初始化标签数据框
  labels <- data.frame(matrix(NA, nrow = nrow(Condition), ncol = ncol(codes)))
  names <- rep(NA, ncol(codes))
  
  # 循环匹配编码并生成标签列
  for (i in seq_len(ncol(codes))) {
    # 统一数据类型后匹配
    codes_col <- as.character(codes[, i])
    labels[, i] <- CodeConditionPreg$label[match(codes_col, CodeConditionPreg$Code)]
    names[i] <- paste0("Other_Complications", i-1)
  }
  
  # 重命名第一列为Complications
  names[1] <- "Complications"
  colnames(labels) <- names
  
  # 合并原数据与标签列
  Condition <- cbind(Condition, labels)
  
  # 返回处理后的数据集
  return(Condition)
}

# 测试函数
Birth_IndicatorsWithConditions <- CleanConditionPreg(Birth_IndicatorsConditions)

# 查看结果
print(Birth_IndicatorsWithConditions)

关键修改点 #

• 修正列名引用，将CodeConditionPreg$code改为CodeConditionPreg$Code，匹配正确的列名。
• 统一编码列的数据类型为字符型，避免因类型不一致导致的匹配失败。
• 添加return(Condition)语句，确保函数返回处理后的数据集。
• 使用seq_len()替代seq(1, ncol(codes))，代码逻辑更严谨。

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