如何编写R函数批量运行不同因变量的同类多元回归并存储结果为列表

核心问题根源 #

• 原有代码直接传入因变量的数值向量，lm()无法识别实际因变量名，导致公式显示异常
• sapply默认会自动简化输出结果结构，导致lm对象的格式被破坏，无法正常调用summary()等方法

最优实现方案 #

推荐通过动态构建公式+返回列表对象的方式实现，代码如下：

# 1. 准备原始数据
my_data <- data.frame(x1=(1:10) + rnorm(10, 3, 1.5), x2=25/3 + rnorm(10, 0, 1), 
                      dep.var1=seq(5, 28, 2.5), dep.var2=seq(100, -20, -12.5), 
                      dep.var3=seq(1, 25, 2.5))

# 2. 定义需要批量计算的因变量名称（字符向量即可，无需提前提取数值）
dep_var_names <- c("dep.var1", "dep.var2")

# 3. 自定义建模函数，动态生成回归公式
fit_lm <- function(dep_name, data) {
  reg_formula <- as.formula(paste0(dep_name, " ~ x1 + x2"))
  lm(reg_formula, data = data)
}

# 4. 批量建模，用lapply保证返回列表结构
model_list <- lapply(dep_var_names, fit_lm, data = my_data)

# 5. 给列表元素命名，方便后续调用
names(model_list) <- paste0("model", seq_along(model_list))

使用方法 #

你可以直接通过列表索引调用对应模型，所有lm对象的原生方法都支持：

# 查看dep.var1的回归结果
summary(model_list$model1)
# 提取dep.var2的回归系数
coef(model_list$model2)

此时回归结果的公式会正常显示为dep.var1 ~ x1 + x2的完整格式。

适配原有写法的方案 #

如果你需要保留原有的传入因变量数值列表的逻辑，可以用mapply同时传入数值和对应名称：

# 原有因变量列表
dep.var <- list(dep.var1=my_data$dep.var1, dep.var2=my_data$dep.var2)

fit_lm2 <- function(dep_vec, dep_name, data) {
  reg_formula <- as.formula(paste0(dep_name, " ~ x1 + x2"))
  fit <- lm(reg_formula, data = data)
  fit$call$formula <- reg_formula
  return(fit)
}

model_list <- mapply(fit_lm2, dep.var, names(dep.var), MoreArgs = list(data = my_data), SIMPLIFY = FALSE)
names(model_list) <- paste0("model", seq_along(model_list))

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