将自定义ggplot图例添加到虚线和置信带中

debugcn 发表于 Dev

强尼夫

我在为自定义带和虚线设置自定义图例时遇到麻烦。到目前为止，这是我的图表。

di<-matrix(ncol = 3,nrow = 5) %>% as.data.frame()
colnames(di)<-c('group','estimate','SE')
di<-di %>% mutate(group=1:5,
                  estimate=c(0.5,9.6,13,15,23.1),
                  SE=14)

ggplot(di, aes(x=group, y=estimate)) +
  geom_point() + 
  geom_errorbar(width=.5, aes(ymin=estimate-(1.647*SE), ymax=estimate+(1.647*SE)), colour="black") +
  xlab('Group') +
  ylab('Treatment Effect') +
  labs(title="GATE with confidence bands", 
       subtitle="Point estimates and confidence bands are derived using median of all splits") +
  geom_hline(yintercept=c(7.83,22.55), 
             linetype="longdash",
             col='darkred') +
  geom_hline(yintercept=15.19, 
             linetype="longdash",
             col='blue')

看起来像这样：

但是，我希望它看起来像这样，具有完全相同的图例：

有什么建议吗？

斯蒂芬

可以这样实现：

作为一般规则：如果你想有一个传说，你必须映射美学的东西，例如招color=...入aes()了所有四个geoms
然后可以通过设置所需的颜色值 scale_color_manual
对于geom_hline我们也有通过yintercept为aes()过。为此，这些得到具有所需值的辅助数据帧。
为了固定图例中的线条和形状，我使用guide_legend的overide.aes来删除图例中不想要的点以及删除该点的线条。此外，我将图例的行数设置为2。
可以通过labels和的breaks参数设置图层的标签和顺序scale_color_manual
在左上角移动图例，并通过theme选项摆脱图例和按键的背景填充。

library(ggplot2)

di <- data.frame(
  group = 1:5,
  estimate = c(0.5, 9.6, 13, 15, 23.1),
  SE = 14
)

labels <- c(point = "Point", error = "Error", blue = "Blue", darkred = "Red")
breaks <- c("blue", "darkred", "point", "error")

ggplot(di, aes(x = group, y = estimate)) +
  geom_point(aes(color = "point"), size = 3) +
  geom_errorbar(width = .5, aes(
    ymin = estimate - (1.647 * SE),
    ymax = estimate + (1.647 * SE),
    color = "error"
  )) +
  scale_color_manual(values = c(
    point = "black",
    error = "black",
    blue = "blue",
    darkred = "darkred"
  ), labels = labels, breaks = breaks) +
  labs(
    title = "GATE with confidence bands",
    subtitle = "Point estimates and confidence bands are derived using median of all splits",
    x = "Group",
    y = "Treatment Effect",
    color = NULL, linetype = NULL, shape = NULL
  ) +
  geom_hline(
    data = data.frame(yintercept = c(7.83, 22.55)),
    aes(yintercept = yintercept, color = "darkred"), linetype = "longdash"
  ) +
  geom_hline(
    data = data.frame(yintercept = 15.19),
    aes(yintercept = yintercept, color = "blue"), linetype = "longdash"
  ) +
  guides(color = guide_legend(override.aes = list(
    shape = c(NA, NA, 16, NA),
    linetype = c("longdash", "longdash", "blank", "solid")
  ), nrow = 2, byrow = TRUE)) +
  theme(legend.position = c(0, 1), 
        legend.justification = c(0, 1), 
        legend.background = element_rect(fill = NA),
        legend.key = element_rect(fill = NA))