如何为sjPlot::plot_models网格图按系数正负设置颜色？

要让sjPlot::plot_models绘制的图中，正系数显示蓝色、负系数显示红色，可以用以下两种方法实现：

方法一：自定义颜色向量 #

先提取所有模型的系数值，根据正负生成对应的颜色向量，再传递给plot_models的colors参数：

set.seed(10)
a <- sample(c("low", "medium", "high"), 100, replace = T)
b <- rnorm(100)
c <- rnorm(100)

mod1 <- lm(b ~ a)
mod2 <- lm(c ~ a)

# 提取两个模型的非截距项系数，判断正负生成颜色向量
coef_values <- c(coef(mod1)[-1], coef(mod2)[-1])
color_list <- ifelse(coef_values > 0, "blue", "red")

# 传入colors参数绘制图形
sjPlot::plot_models(mod1, mod2, grid = T, colors = color_list)

方法二：修改ggplot对象的颜色映射 #

plot_models返回的是ggplot对象，可直接通过scale_color_manual修改颜色映射，基于系数的正负值设置颜色：

set.seed(10)
a <- sample(c("low", "medium", "high"), 100, replace = T)
b <- rnorm(100)
c <- rnorm(100)

mod1 <- lm(b ~ a)
mod2 <- lm(c ~ a)

# 生成基础图形
plot_obj <- sjPlot::plot_models(mod1, mod2, grid = T)

# 根据系数正负调整颜色
plot_obj + ggplot2::scale_color_manual(
  values = ifelse(plot_obj$data$estimate > 0, "blue", "red")
)

两种方法都能实现需求，第一种更直接，第二种利用ggplot的灵活性，适合需要后续调整其他图层的场景。

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