如何在reghelper包的graph_model函数中添加图表标题、标签及图例

reghelper包的graph_model()函数基于ggplot2生成图表，因此可以直接通过链式调用ggplot2的标签函数来实现需求。以下是修改后的完整代码：

dv1 = runif(n = 100, min = 1, max = 7)
dv2 = runif(n = 100, min = 1, max = 7)
dv3 = runif(n = 100, min = 1, max = 7)
country <- rep(c("India", "US", "Poland"), length.out = 100)
df <- data.frame(country, dv1, dv2, dv3)
library(reghelper)
library(ggplot2) # 需加载ggplot2以使用标签函数

dv1 <- as.numeric(dv1)
dv2 <- as.numeric(dv2)
dv3 <- as.numeric(dv3)
country <- as.factor(country)

lm0 <- lm(dv1 ~ dv2 * dv3 + country, data = df, na.action = na.exclude)
summary(lm0)

# 添加标题、轴标签和图例标签的调用代码
graph_model(lm0, y=dv1, x=dv2, lines=country, split=dv3) +
  ggtitle("DV1与DV2的关系（按国家分组，DV3拆分）") + # 设置图表标题
  xlab("自变量DV2") + # 设置X轴标签
  ylab("因变量DV1") + # 设置Y轴标签
  labs(color = "国家") # 设置图例标题（对应lines参数的分组）

关键修改说明： #

• 必须加载ggplot2包，因为graph_model()返回的是ggplot对象，依赖ggplot的函数进行标签修改。
• ggtitle()：替换括号内的文本即可自定义图表主标题。
• xlab()和ylab()：分别自定义X轴和Y轴的显示文本。
• labs(color = "国家")：由于使用lines=country分组，图例的颜色映射对应该分组，此参数用于修改图例的标题。

如果需要调整图例位置或其他样式，还可以继续添加theme(legend.position = "bottom")这类ggplot主题函数优化展示效果。

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