如何在R中为多个原始数据框批量添加指定计算列？

你已经通过函数f1实现了给数据框添加新列的逻辑，但使用lapply处理数据框列表时，仅能得到处理后的列表结果，无法直接更新原始数据框。以下是几种可行的解决方案：

方案1：使用list2env将处理结果写回全局环境 #

首先将数据框以命名列表的形式传入lapply（列表名称需与原始变量名一致），处理完成后用list2env将结果赋值回全局环境：

# 构造命名列表，名称与原始数据框变量名匹配
dflist <- list(WM_Purchased_2 = WM_Purchased_2, 
               WM_Purchased_3 = WM_Purchased_3, 
               WM_Purchased_4 = WM_Purchased_4)
# 处理后将结果写回全局环境
list2env(lapply(dflist, f1), envir = .GlobalEnv)

# 对应可复现示例的代码：
listdf <- list(m1 = m1, m2 = m2)
list2env(lapply(listdf, f1), envir = .GlobalEnv)

执行后，m1、m2或WM_Purchased_2/3/4会被直接更新为添加了新列的版本。

方案2：通过循环逐个更新原始数据框 #

如果偏好循环逻辑，可以遍历数据框的变量名，结合get和assign操作全局环境中的对象：

# 定义需要处理的数据框名称向量
df_names <- c("WM_Purchased_2", "WM_Purchased_3", "WM_Purchased_4")
# 循环处理并赋值回原始变量
for (name in df_names) {
  dat <- get(name)
  dat <- f1(dat)
  assign(name, dat, envir = .GlobalEnv)
}

# 对应可复现示例的代码：
df_names <- c("m1", "m2")
for (name in df_names) {
  dat <- get(name)
  dat <- f1(dat)
  assign(name, dat, envir = .GlobalEnv)
}

方案3：改进函数直接修改原始数据框（不推荐） #

可以通过引用传递的方式直接修改全局环境中的数据框，但这种方式会打破R函数的纯性，容易引发意外问题，仅作参考：

f1 <- function(dat_name) {
  # 从全局环境获取数据框
  dat <- get(dat_name, envir = .GlobalEnv)
  dat <- dat %>% 
    mutate(A_Purchase_Freq = 
            ifelse(utfoodb=="Daily", tfoodb*30*mfoodbly, 
            ifelse(utfoodb=="Weekly", tfoodb*4*mfoodbly,
            ifelse(utfoodb=="Monthly", tfoodb*1*mfoodbly,
            ifelse(utfoodb=="Quarterly", tfoodb*4, 
            ifelse(utfoodb=="Bi-Annually", tfoodb*2,
            ifelse(utfoodb=="Annually", tfoodb*1, NA))))))) %>%
    mutate(A_Purchase_Freq= as.numeric(A_Purchase_Freq)) %>%
    mutate(T_A_Expenditure = A_Purchase_Freq * cfoodb) 
  # 将处理后的数据框赋值回全局环境
  assign(dat_name, dat, envir = .GlobalEnv)
}

# 逐个调用函数更新数据框
f1("WM_Purchased_2")
f1("WM_Purchased_3")
f1("WM_Purchased_4")

注：此处将原代码中的字符串"NA"替换为R原生缺失值NA，避免转数值型时出现不必要的类型转换问题。

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