在R中如何循环切换回归指示变量的省略基准类别并自动绘图？

核心思路 #

要自动切换基准类别，本质是动态修改因子变量的参考水平，然后循环遍历所有基准组合，依次拟合模型并生成交互图。以下是可直接运行的实现代码，包含完整逻辑：

完整代码实现 #

# 1. 先确保变量是因子类型（如果还不是的话）
data$`10indicators` <- as.factor(data$`10indicators`)
data$`3indicators` <- as.factor(data$`3indicators`)

# 2. 获取两个变量的所有类别
cat_10 <- levels(data$`10indicators`)
cat_3 <- levels(data$`3indicators`)

# 3. 生成所有基准组合（10*3=30种组合）
all_ref_pairs <- expand.grid(ref_10 = cat_10, ref_3 = cat_3, stringsAsFactors = FALSE)

# 4. 循环遍历所有组合，拟合模型并绘图
for (i in 1:nrow(all_ref_pairs)) {
  # 设置当前循环的基准类别
  current_ref_10 <- all_ref_pairs$ref_10[i]
  current_ref_3 <- all_ref_pairs$ref_3[i]
  
  # 修改因子的参考水平
  data$`10indicators` <- relevel(data$`10indicators`, ref = current_ref_10)
  data$`3indicators` <- relevel(data$`3indicators`, ref = current_ref_3)
  
  # 拟合logistic回归模型
  mylogit1 <- glm(MyDepVar ~ `10indicators` * `3indicators`, data = data, family = "binomial")
  
  # 生成带标题的交互图（标题显示当前基准组合）
  plot_title <- paste0("Omitted category: ", current_ref_10, "/", current_ref_3)
  p <- plot_model(mylogit1, type = "int", title = plot_title, show.legend = TRUE,
                  transform = "plogis", show.values = TRUE, dot.size = 5, line.size = 1.5, 
                  value.offset = .3, colors = "Paired") + theme_sjplot()
  
  # 可选：打印模型摘要（如果需要查看结果）
  # print(summary(mylogit1))
  
  # 可选：保存图片到本地（按基准组合命名）
  ggsave(filename = paste0("interaction_plot_", current_ref_10, "_", current_ref_3, ".png"),
         plot = p, width = 10, height = 8, dpi = 300)
  
  # 显示图形（如果在RStudio中运行）
  print(p)
}

关键细节说明 #

• 使用relevel()函数动态修改因子的参考水平，这是切换基准类别的核心操作。
• expand.grid()生成所有可能的基准组合，确保覆盖10*3=30种情况。
• 绘图标题自动标注当前使用的基准类别，方便区分不同图形；保存图片时也用基准类别命名，避免文件覆盖。
• 如果不需要保存图片，可删除ggsave()部分；如果不需要打印模型摘要，可注释掉print(summary(mylogit1))。

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