如何用R语言自定义函数给指定变量名添加前缀

情况1：复制原变量并添加前缀（保留原变量） #

你原来的代码问题在于直接使用data$demo_x时，R会把demo_x当成固定的变量名，不会解析你传入的参数x。需要用字符串拼接生成新变量名，再通过[[ ]]来操作数据框的列：

# 定义添加前缀的函数
add_demo_prefix <- function(var_name) {
  # 拼接新变量名
  new_var <- paste0("demo_", var_name)
  # 复制原变量到新变量
  data[[new_var]] <- data[[var_name]]
}

# 对目标变量调用函数
add_demo_prefix("age")
add_demo_prefix("gender")

执行后，你的数据集会新增demo_age和demo_gender两个变量，原age、gender以及zipcode、addresses都会保留。

情况2：直接重命名原变量（替换原变量名） #

如果不需要保留原变量名，想直接把age改成demo_age、gender改成demo_gender，可以用基础R的列名替换方法，更高效：

# 指定需要修改的变量
target_vars <- c("age", "gender")
# 生成新的变量名
new_names <- paste0("demo_", target_vars)
# 找到目标变量在数据框中的位置
var_index <- match(target_vars, colnames(data))
# 替换列名
colnames(data)[var_index] <- new_names

如果习惯用tidyverse工具包，也可以用dplyr的rename_with函数：

library(dplyr)

data <- data %>%
  rename_with(~ paste0("demo_", .x), all_of(target_vars))

为什么你的原代码不生效？ #

data$x中的x是字面量，R不会把它当成你传入的函数参数"age"或"gender"。而data[[x]]可以接受字符串作为索引，能正确指向你要操作的变量；同理，新变量名也需要通过paste0拼接成字符串，再用[[new_var]]赋值。

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