R语言：如何无需多次调用assign在父环境中更新变量？

问题 #

给定如下R代码：

f_parent <- function() {
 a1 <- 10
 a2 <- 20
 # .....
 a100 <- 100

 env <- environment()

 f_child <- function(a1, a2, ..., a100, env) {
  a1 <- a1 * 3
  a2 <- a2 * 5
  # .....
  a100 <- a100 * 1837

  assign('a1', a1, envir = env)
  assign('a2', a2, envir = env)
  # .....
  assign('a100', a100, envir = env)
 }
# .....
}

是否可以避免在f_child函数内多次调用assign，从而在f_parent环境中更新a1至a100变量？

解决方法 #

当然可以，这里提供两种实用的批量更新方式：

方法一：用列表批量处理+list2env赋值 #

把需要更新的变量先打包成列表，子函数处理完列表后，一次性将列表内容同步到父环境，彻底省去多次assign：

f_parent <- function() {
  # 初始化变量并存入列表，后续也能直接在环境中使用这些变量
  vars <- list(a1 = 10, a2 = 20, a100 = 100)
  list2env(vars, environment())
  
  env <- environment()
  
  f_child <- function(vars, env) {
    # 批量修改列表中的变量值
    vars$a1 <- vars$a1 * 3
    vars$a2 <- vars$a2 * 5
    vars$a100 <- vars$a100 * 1837
    
    # 一次性把所有变量同步回父环境
    list2env(vars, envir = env)
  }
  
  # 调用子函数验证
  f_child(vars, env)
  print(a1)  # 输出30
  print(a2)  # 输出100
  print(a100) # 输出183700
}

f_parent()

方法二：循环遍历变量名批量处理 #

如果变量名有规律（比如都是a+数字），可以先定义变量名和对应乘数的向量，通过循环批量完成取值、计算、赋值：

f_parent <- function() {
  a1 <- 10
  a2 <- 20
  a100 <- 100
  
  env <- environment()
  # 定义要处理的变量名和对应乘数
  var_names <- paste0("a", c(1, 2, 100))
  multipliers <- c(3, 5, 1837)
  
  f_child <- function(var_names, multipliers, env) {
    # 循环处理每个变量
    for(i in seq_along(var_names)) {
      current_val <- get(var_names[i], envir = env)
      new_val <- current_val * multipliers[i]
      assign(var_names[i], new_val, envir = env)
    }
  }
  
  # 调用子函数验证
  f_child(var_names, multipliers, env)
  print(a1)  # 30
  print(a2)  # 100
  print(a100) # 183700
}

f_parent()

两种方法里，用列表的方式更简洁高效，也更符合R的函数式编程习惯；循环方式适合变量数量多但命名有规律的场景。

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