R语言函数传递数据框列索引参数问题求助

原函数的核心问题是：传入数字索引时，~x会把x当作常量数值（比如传入2时，所有观测的自变量都是2），而非引用ChickWeight数据框对应索引的列；同时attach()容易引发环境变量冲突，建议避免使用。以下是修正后的方案：

修正后的函数 #

library(gtsummary)
library(dplyr)

RTable <- function(col) {
  # 兼容列名/列索引输入，统一转为列名
  col_name <- if (is.numeric(col)) {
    colnames(ChickWeight)[col]
  } else {
    as.character(substitute(col))
  }
  
  # 动态构造回归公式
  formula <- reformulate(col_name, response = "weight")
  
  # 拟合模型并生成回归表格
  glm(formula, data = ChickWeight) %>%
    tbl_regression(exponentiate = FALSE)
}

关键逻辑说明 #

• 兼容两种输入类型：数字索引通过colnames(ChickWeight)[col]转成对应列名；直接传入列名（如Time）时，用substitute(col)提取变量名再转字符。
• 动态构造公式：用reformulate()生成回归公式，确保模型正确引用数据框列，同时表格会显示真实列名（而非x）。
• 移除attach()：直接在glm中指定data=ChickWeight，避免环境变量混乱。

调用示例 #

用列名调用

r1 <- RTable(Time)
r2 <- RTable(Diet)
merge_d <- tbl_stack(tbls = list(r1, r2))
merge_d

用列索引调用

g1 <- RTable(2)  # 对应Time列
g2 <- RTable(3)  # 对应Diet列
merge_gd <- tbl_stack(tbls = list(g1, g2))
merge_gd  # 结果与merge_d完全一致

循环批量处理

# 批量处理第2、3列
col_indices <- c(2, 3)
results_list <- lapply(col_indices, RTable)
merge_all <- tbl_stack(results_list)
merge_all

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