在R语言中使用循环为已有DataFrame创建列的问题求助

嘿，我来帮你搞定这个循环的问题！你的思路完全正确——用循环替代重复的复制粘贴代码太明智了，不过这里有两个关键的小问题导致循环没按预期工作，还有可以大幅简化代码的方法：

问题出在哪？ #

• 动态列名和列引用错误：你写的Large_Data$column_k会直接创建一个叫column_k的固定列名，而不是跟着循环变量k生成column_a 1、column_b 1这类对应列名；同样Large_Data$k``其实是在找名为k的列，而不是循环里的"a 1"、"b 1"这些目标列。得用R的[[操作符来动态处理列名才行。
• 嵌套ifelse过于冗余：这么多层嵌套不仅难读，还容易漏写括号出错，换成映射表的方式会简洁很多。

修复后的基础R循环版本 #

先给你改好能直接用的循环代码，解决列名的问题：

names <- c("a 1", "b 1", "c 1", "d 1")

for (k in names) {
  # 动态生成新列名，比如把"a 1"转为"column_a 1"
  new_col_name <- paste0("column_", k)
  # 用[[操作符动态引用原列和赋值新列
  Large_Data[[new_col_name]] <- ifelse(Large_Data[[k]] == "I-2", 2, 
                                       ifelse(Large_Data[[k]] == "I-3", 3, 
                                              ifelse(Large_Data[[k]] == "I-4", 4, 
                                                     ifelse(Large_Data[[k]] == "I-5", 5, 
                                                            ifelse(Large_Data[[k]] == "I-6", 6, 
                                                                   ifelse(Large_Data[[k]] == "I-7", 7, 
                                                                          ifelse(Large_Data[[k]] == "S-1", 8, 
                                                                                 ifelse(Large_Data[[k]] == "S-2", 9, 
                                                                                        ifelse(Large_Data[[k]] == "S-3", 10, 
                                                                                               ifelse(Large_Data[[k]] == "S-4", 11, 
                                                                                                      ifelse(Large_Data[[k]] == "P-1", 12, 
                                                                                                             ifelse(Large_Data[[k]] == "P-2", 13, 
                                                                                                                    ifelse(Large_Data[[k]] == "P-3", 14, 
                                                                                                                           ifelse(Large_Data[[k]] == "D-1", 15, 
                                                                                                                                  ifelse(Large_Data[[k]] == "D-2", 16, 99)))))))))))))))
}

更简洁的优化版本（推荐） #

用映射表替代嵌套ifelse，代码可读性和维护性都更强：

# 先定义值的映射关系，一目了然
value_map <- c(
  "I-2" = 2, "I-3" = 3, "I-4" = 4, "I-5" = 5, "I-6" = 6, "I-7" = 7,
  "S-1" = 8, "S-2" = 9, "S-3" = 10, "S-4" = 11,
  "P-1" = 12, "P-2" = 13, "P-3" = 14,
  "D-1" = 15, "D-2" = 16
)

names <- c("a 1", "b 1", "c 1", "d 1")

for (k in names) {
  new_col_name <- paste0("column_", k)
  # 用映射表匹配值，未匹配到的设为99
  mapped_values <- value_map[Large_Data[[k]]]
  Large_Data[[new_col_name]] <- ifelse(is.na(mapped_values), 99, mapped_values)
}

如果你用tidyverse（更高效） #

如果平时用dplyr这类工具，甚至可以不用循环，用across一次性处理所有列：

library(dplyr)

value_map <- c(
  "I-2" = 2, "I-3" = 3, "I-4" = 4, "I-5" = 5, "I-6" = 6, "I-7" = 7,
  "S-1" = 8, "S-2" = 9, "S-3" = 10, "S-4" = 11,
  "P-1" = 12, "P-2" = 13, "P-3" = 14,
  "D-1" = 15, "D-2" = 16
)

Large_Data <- Large_Data %>%
  mutate(across(all_of(names), 
                ~ifelse(is.na(value_map[.]), 99, value_map[.]), 
                .names = "column_{.col}"))

这里across会自动遍历你指定的列，.names参数帮你自动生成带前缀的新列名，非常省心！

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