如何在R中绘制连续变量交互效应的指定样式可视化图？

你需要展示连续变量X1对Y的影响依赖于X2的交互效应，以下是基于你的R代码的两种实现方案，包括你打算用的segments()方法，以及更便捷的ggplot2替代方案：

原始模拟与建模代码 #

X1 <- rnorm(1000,0,1)
X2 <- rnorm(1000,0,1)
error <- rnorm(1000,0,0.5)
intercept <- 5
coef_1 <- 0.5
coef_2 <- 1
coef_3 <- -1.5  
Y <- intercept + coef_1*X1 + coef_2*X2 + coef_3*X1*X2 + error
data <- data.frame(Y=Y,X1=X1,X2=X2)
fit <- lm(Y ~ X1*X2,data=data)
summary(fit)

方案1：使用segments()实现基础绘图 #

这种方法手动控制直线绘制，适合高度自定义细节：

# 提取模型系数
coefs <- coef(fit)
intercept <- coefs[1]
b1 <- coefs[2]
b2 <- coefs[3]
b3 <- coefs[4]

# 选择X2的代表性取值：均值、均值±1标准差（连续变量交互可视化常用做法）
x2_vals <- with(data, c(mean(X2)-sd(X2), mean(X2), mean(X2)+sd(X2)))
x1_range <- range(data$X1)

# 初始化散点图
plot(Y ~ X1, data = data, pch = 16, col = rgb(0,0,0,0.2), 
     main = "X1对Y的交互效应随X2变化", xlab = "X1", ylab = "Y")

# 循环添加不同X2水平下的预测直线
for(x2 in x2_vals){
  y_start <- intercept + b1*x1_range[1] + b2*x2 + b3*x1_range[1]*x2
  y_end <- intercept + b1*x1_range[2] + b2*x2 + b3*x1_range[2]*x2
  segments(x0 = x1_range[1], y0 = y_start, x1 = x1_range[2], y1 = y_end, 
           lwd = 2, col = ifelse(x2 == mean(data$X2)-sd(X2), "blue", 
                                 ifelse(x2 == mean(data$X2), "red", "green")))
}

# 添加图例说明X2取值
legend("topright", legend = paste0("X2 = ", round(x2_vals, 2)), 
       col = c("blue", "red", "green"), lwd = 2)

方案2：使用ggplot2快速生成可视化 #

ggplot2能更高效生成美观且易维护的交互效应图：

library(ggplot2)

# 构建预测数据集
x1_seq <- seq(min(data$X1), max(data$X1), length.out = 100)
x2_vals <- with(data, c(mean(X2)-sd(X2), mean(X2), mean(X2)+sd(X2)))
pred_data <- expand.grid(X1 = x1_seq, X2 = x2_vals)
pred_data$Y_pred <- predict(fit, newdata = pred_data)

# 绘制交互效应图
ggplot(data, aes(x = X1, y = Y)) +
  geom_point(alpha = 0.2, color = "gray40") +
  geom_line(data = pred_data, aes(y = Y_pred, color = factor(X2)), linewidth = 1.2) +
  scale_color_discrete(labels = paste0("X2 = ", round(x2_vals, 2))) +
  labs(title = "X1对Y的交互效应随X2变化", x = "X1", y = "Y", color = "X2取值") +
  theme_minimal()

两种方案都通过选取X2的代表性水平（均值±1标准差），展示不同X2下X1对Y的预测斜率变化，清晰呈现交互效应。

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