ggplot：使用不同的线型将回归线延伸到预测值

Question

是否有一种简单的方法可以将虚线从实线回归线的末端延伸到预测值？

以下是我的基本尝试：

x = rnorm(10)
y = 5 + x + rnorm(10,0,0.4)

my_lm <- lm(y~x)
summary(my_lm)

my_intercept <- my_lm$coef[1]
my_slope <- my_lm$coef[2]
my_pred = predict(my_lm,data.frame(x = (max(x)+1)))

ggdf <- data.frame( x = c(x,max(x)+1), y = c(y,my_pred), obs_Or_Pred = c(rep("Obs",10),"Pred") )

ggplot(ggdf, aes(x = x, y = y, group = obs_Or_Pred ) ) +
     geom_point( size = 3, aes(colour = obs_Or_Pred) ) + 
     geom_abline( intercept = my_intercept, slope = my_slope, aes( linetype = obs_Or_Pred ) )

这并没有给出我希望看到的输出。我在SO上看了一些其他的答案，并没有看到任何简单的事情。我想出的最好的是：

ggdf2 <- data.frame( x = c(x,max(x),max(x)+12), y = c(y,my_intercept+max(x)*my_slope,my_pred), obs_Or_Pred = c(rep("Obs",8),"Pred","Pred"), show_Data_Point = c(rep(TRUE,8),FALSE,TRUE) )

ggplot(ggdf2, aes(x = x, y = y, group = obs_Or_Pred ) ) +
     geom_point( data = ggdf2[ggdf2[,"show_Data_Point"],] ,size = 3, aes(colour = obs_Or_Pred) ) + 
     geom_smooth( method = "lm", se=F, aes(colour = obs_Or_Pred, linetype=obs_Or_Pred) )

这给出了正确的输出，但我必须包含一个额外的列，指定我是否要显示数据点。如果我不这样做，我最终得到这两个图中的第二个图，它在拟合回归线的末尾有一个额外的点：

是否有更简单的方法告诉ggplot预测线性模型中的单个点并绘制虚线？

Answer 1

您可以仅使用实际数据绘制点，并构建预测数据框以添加线条。请注意，max(x)出现两次，因此它可以是Obs行和Pred行的终结点。我们还使用了shape美学，以便我们可以删除否则会出现在Pred的图例键中的点标记。

# Build prediction data frame
pred_x = c(min(x),rep(max(x),2),max(x)+1)
pred_lines = data.frame(x=pred_x,
                        y=predict(my_lm, data.frame(x=pred_x)),
                        obs_Or_Pred=rep(c("Obs","Pred"), each=2))

ggplot(pred_lines, aes(x, y, colour=obs_Or_Pred, shape=obs_Or_Pred, linetype=obs_Or_Pred)) +
  geom_point(data=data.frame(x,y, obs_Or_Pred="Obs"), size=3) +
  geom_line(size=1) +
  scale_shape_manual(values=c(16,NA)) +
  theme_bw()

Answer 2

半丑：您可以使用scale_x_continuous(limits =设置用于预测的x值范围。首先用fullrange = TRUE绘制预测线，然后在顶部添加“观察”线。请注意，过度绘制不能完美呈现，您可能需要稍微增加观察线的大小。

ggplot(d, aes(x, y)) +
  geom_point(aes(color = "obs")) +
  geom_smooth(aes(color = "pred", linetype = "pred"), se = FALSE, method = "lm",
                                                      fullrange = TRUE) +
  geom_smooth(aes(color = "obs", linetype = "obs"), size = 1.05, se = FALSE, method = "lm") +
  scale_linetype_discrete(name = "obs_or_pred") +
  scale_color_discrete(name = "obs_or_pred") +
  scale_x_continuous(limits = c(NA, max(x) + 1))

但是，我倾向于同意Gregor：“ggplot是一个绘图包，而不是建模包。”