在 ggplot 中，如何在散点图中绘制两组的平均线

Question

我想在散点图中显示两组的平均值。我已经对数据进行了排序，因此这些组彼此相邻。第 1 组是前 11 条记录，第 2 组是接下来的 133 条记录。我如何告诉 ggplot 在第一组（第 1-11 号）的范围内画一条线，为第二组（第 12-133 号）画一条线。

这是我目前所拥有的：

代码在这里：

library(tidyverse)
library(tidymodels)

data(ames)
ames <- AmesHousing::make_ames()

set.seed(1)
split  <- initial_split(ames, prop = 0.95, strata = "Sale_Price")
ames_plot   <- testing(split)

model1 <- lm(Sale_Price ~ Central_Air, data = ames_plot)

p1 <- model1 %>%
  broom::augment() %>%
  arrange(Central_Air) %>% 
  mutate(House = row_number()) %>% 
  ggplot(aes(House, Sale_Price, color = Central_Air)) + 
  geom_point(size = 1, alpha = 0.3) +
  geom_segment(aes(x = 1, y = .fitted, xend = 144, yend =.fitted)) +
  scale_y_continuous(labels = scales::dollar) 
p1

使用 geom_smooth(formula = 'y ~ x', se = FALSE, method = "lm") 而不是 geom_segment() 让我接近我想要的，但我想显示来自 lm() 的实际预测值。

Answer 1

最好只是总结该层的数据。例如

model1 %>%
  broom::augment() %>%
  arrange(Central_Air) %>% 
  mutate(House = row_number()) %>% 
  ggplot(aes(House, Sale_Price, color = Central_Air)) + 
  geom_point(size = 1, alpha=.3) +
  geom_segment(aes(x = first, y = .fitted, xend = last, yend =.fitted), 
    data = function(x) {
      x %>% 
        group_by(Central_Air)  %>% 
        summarize(first=first(House), last=last(House), .fitted=mean(.fitted), .groups="drop_last")
  }) + 
  scale_y_continuous(labels = scales::dollar)