r - 如何在黄土和样条回归中添加协变量，然后使用 ggplot2 在 r 中绘制它

Question

我知道如何仅用一个自变量绘制黄土和样条回归。

library(tidyverse)

# loess
ggplot(mtcars, aes(x = mpg, y = drat)) + 
  geom_point() + 
  geom_smooth(method = 'loess', formula = y ~ x)

# splines
ggplot(mtcars, aes(x = mpg, y = drat)) + 
  geom_point() + 
  geom_smooth(method = 'lm', formula = y ~ splines::bs(x, 8))

让我卡住的是，我有多个自变量，例如 x1、x2、x3，我创建了一个这样的模型:y ~ x1 + x2 + x3，我只想用 loess 或 spline 绘制 y 和 x1 之间的曲线。

我尝试过但失败了。

cyl 和 gear 是模型中的协变量，我只对 drat 和 mpg 感兴趣

# loess
ggplot(mtcars, aes(x = mpg, y = drat)) + 
  geom_point() + 
  geom_smooth(method = 'loess', formula = y ~ x + cyl + gear)
# splines
ggplot(mtcars, aes(x = mpg, y = drat)) + 
  geom_point() + 
  geom_smooth(method = 'lm', formula = y ~ splines::bs(x, 8)+ cyl + gear)

我们将不胜感激任何帮助。

Answer 1

一种方法是分别拟合 loess 并绘制具有置信区间的拟合线。

fit = predict(loess(drat ~ mpg + cyl + gear, data=mtcars, span=0.5), se=T)
#tem = predict(loess(drat~ mpg,data=mtcars), se=T)

dat = mtcars
dat$loess = fit$fit
dat$ymax  = fit$fit + fit$se.fit * abs(qnorm((1-0.95)/2))
dat$ymin  = fit$fit - fit$se.fit * abs(qnorm((1-0.95)/2))

ggplot(dat, aes(x = mpg, y = drat)) + 
  geom_point() + 
  geom_line(aes(y=loess), color='blue', size=1) +
  geom_ribbon(aes(ymin=ymin, ymax=ymax), alpha=0.2)