如何编写R函数，基于指定列创建/替换iris数据集的目标列？

我有iris数据集，想写一个函数，通过合并指定列来创建或替换目标列。函数需要两个参数：目标列的名称，以及需要合并的列名列表。

比如我想合并Petal.Width和Species列，期望实现的效果等同于执行：

iris$Petal.Width <- paste0(iris$Petal.Width, iris$Species)

我尝试了以下代码，但不清楚如何处理动态列名：

func <- function(data, names, column_names){
data <- data %>% mutate_at(vars(column_names), names = paste0(column_names))
}

iris <- func(iris, "Petal.Width", c("Petal.Width","Species"))

请问该如何正确实现这个函数？

解决方案 #

方法1：使用dplyr的最新语法（推荐） #

mutate_at已被dplyr弃用，推荐用across结合动态列名处理，配合do.call实现逐行拼接：

library(dplyr)

merge_cols <- function(data, target_col, merge_cols_list) {
  data %>%
    mutate(
      # 动态指定目标列，拼接指定列的每行内容
      !!sym(target_col) := do.call(paste0, pick(all_of(merge_cols_list)))
    )
}

# 测试：合并Petal.Width和Species到Petal.Width列
iris_updated <- merge_cols(iris, "Petal.Width", c("Petal.Width", "Species"))

方法2：基础R实现（无依赖） #

如果不想使用dplyr，用基础R的apply按行处理即可：

merge_cols_base <- function(data, target_col, merge_cols_list) {
  # 按行提取指定列并拼接
  merged_vals <- apply(data[merge_cols_list], 1, function(row) paste0(row, collapse = ""))
  # 替换或创建目标列
  data[[target_col]] <- merged_vals
  return(data)
}

# 测试
iris_updated_base <- merge_cols_base(iris, "Petal.Width", c("Petal.Width", "Species"))

关键细节 #

• 参数命名：避免使用names这类R内置函数名作为参数，改用target_col（目标列）、merge_cols_list（待合并列列表）更清晰。
• 动态列访问：dplyr中用!!sym(target_col)实现动态列赋值；基础R中用data[[target_col]]替代$，适配变量形式的列名。
• 拼接逻辑：do.call(paste0, ...)会把指定列的每行元素作为参数传递给paste0，实现无分隔符拼接；apply(..., 1, ...)则是逐行处理列值。

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