替换R函数中多个变量名并保存修改后的函数

给定如下R函数：

old_fun <- function(x) {
  VarA <- sum(x)
  VarA <- 2 * VarA
  VarX <- VarA %% 2

 return(list("VarA" = VarA, "VarX" = VarX))
}

需要将函数内的VarA替换为VarB、VarX替换为VarY，得到目标函数：

new_fun <- function(x) {
  VarB <- sum(x)
  VarB <- 2 * VarB
  VarY <- VarB %% 2
  
 return(list("VarB" = VarB, "VarY" = VarY))
}

完成替换后需保存new_fun供后续使用。

尝试了以下代码，但生成的new_fun结果为NULL，同时不清楚如何将函数保存到文件夹：

# Variables to replace
variables_to_replace <- c("VarA", "VarX")
new_variables <- c("VarB", "VarY")

# Replace variables in the function code
modified_code <- deparse(substitute(old_fun))
for (i in seq_along(variables_to_replace)) {
  modified_code <- gsub(variables_to_replace[i], new_variables[i], modified_code)
}

# Create a new function with replaced variables
new_fun <- eval(parse(text = modified_code))

1. 修复变量替换逻辑 #

原代码的核心问题是deparse(substitute(old_fun))仅返回字符串"old_fun"，而非函数的源代码。正确做法是用deparse(old_fun)获取完整函数代码，同时添加单词边界\\b避免部分匹配（防止类似VarA1这类变量被误替换）：

# 定义变量替换映射
var_mapping <- c("VarA" = "VarB", "VarX" = "VarY")

# 获取原函数的源代码文本
fun_source <- deparse(old_fun)

# 批量替换变量名
for (old_var in names(var_mapping)) {
  fun_source <- gsub(paste0("\\b", old_var, "\\b"), var_mapping[old_var], fun_source)
}

# 替换函数名old_fun为new_fun
fun_source <- gsub("old_fun", "new_fun", fun_source)

# 解析并生成新函数
new_fun <- eval(parse(text = fun_source))

验证新函数功能：

new_fun(c(1,2,3))
# 输出结果：$VarB [1] 12; $VarY [1] 0，符合预期

2. 保存函数到文件夹 #

有两种常用保存方式：

方式一：保存为RData文件（适合快速加载对象） #

# 创建输出文件夹（如果不存在）
dir.create("output", showWarnings = FALSE)
# 保存函数到指定路径
save(new_fun, file = file.path("output", "new_fun.RData"))

# 后续加载使用
load(file.path("output", "new_fun.RData"))

方式二：保存为R脚本文件（适合查看和编辑代码） #

# 保存函数代码到R脚本
dump("new_fun", file = file.path("output", "new_fun.R"))

# 后续加载使用
source(file.path("output", "new_fun.R"))

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