ggplot：geom_vline 和 geom_hline 的单独图例

Posted 2023-02-16

【中文标题】ggplot：geom_vline 和 geom_hline 的单独图例

【英文标题】：ggplot: Separate legend for both a geom_vline and geom_hline

【发布时间】：2021-09-24 09:45:20

【问题描述】：

我正在尝试在带有构面的 ggplot 中为 geom_vline 和 geom_hline 制作单独的图例。

这是我的代码：

set.seed(1)
vec1 = rnorm(20)
set.seed(2)
vec2 = rnorm(20)
set.seed(3)
vec3 = rnorm(20)

vec  = c(vec1,vec2,vec3)
let  = rep(sample(letters[1:3],20,replace=TRUE),3)
num  = c(rep(1:20,3))
lab  = c(rep("label 1",20),rep("label 2",20),rep("label 3",20))

# Vertical lines for geom_vline
line1 = seq(1,20,5)

df   = data.frame(vector = vec, index = num, name = let, label = lab)

# Define facet order
df$label_f = factor(df$label, levels=c("label 1","label 2","label 3"))

# Dataframe for horizontal lines
hl = data.frame(label = c("label 1","label 2","label 3"), Z = c(0.5,1,-0.5))

# Factor to seperate horizontal lines for facets
hl$label_f = factor(hl$label, levels=c("label 1","label 2","label 3"))

ggplot(data=df,aes(index,vector)) + 
geom_point(aes(alpha = name, color=name)) +
geom_vline(xintercept=line1, color="blue") +
geom_hline(data = hl, aes(yintercept=Z),color="red") +
scale_alpha_manual(name="Indices",
                     labels=c("legend label 1","legend label 2","legend label 3"),
                     values=c(1,0.5,0.5)) +
scale_color_manual(name="Indices",
                     labels=c("legend label 1","legend label 2","legend label 3"),
                     values=c("black","orange","darkorange4")) +        
xlab("index") + ylab("rnorm") +  
facet_wrap(~label_f,ncol=1,scale="free_y") +
theme(legend.position="bottom")

这是ggplot：

我想要一个带有适当标签和颜色的 geom_hline 和 geom_vline 单独的图例（“图例 2”）：

我希望 geom_hline 在图例中是水平的，geom_vline 也是垂直的！

它应该是这样的：

谢谢！

【参考方案1】：

一种选择是使用fill=name 作为geom_point() 的“颜色”，而不是color=name。

另外：

设置pch=21 为 vline_data 创建一个小数据框 将颜色（例如，红色和蓝色）移动到 vline 和 hline aes()

（我还增加了geom_points的大小）

vline_data = data.frame(label_f = c(rep("label 1",length(line1)),rep("label 2",length(line1)),rep("label 3",length(line1))), Z = line1)

ggplot(data=df,aes(index,vector)) + 
  geom_point(aes(fill=name ), size=4,pch=21) +
  geom_vline(data = vline_data, aes(xintercept=Z, color="blue")) +
  geom_hline(data = hl, aes(yintercept=Z, color="red")) +
  scale_fill_manual(name="Indices",
                    labels=c("legend label 1","legend label 2","legend label 3"),
                    values=c("black","orange","darkorange4")
                    ) +
  scale_alpha_manual(name="Indices",
                    labels=c("legend label 1","legend label 2","legend label 3"),
                    values=c(1,0.5,0.5)) +
  scale_color_manual(name="Additional Lines",
                     labels=c("Vertical", "Horizontal"),
                     values=c("blue", "red")) +        
  xlab("index") + ylab("rnorm") +  
  facet_wrap(~label_f,ncol=1,scale="free_y") +
  theme(legend.position="bottom") + 
  guides(alpha="none")

【讨论】：

它不太奏效，因为在图例中线条变成了十字“+”，但其他一切都很好。感谢您的输入，谢谢！

【参考方案2】：

利用ggnewscale 和自定义draw_key 函数可以这样实现：

可以通过ggnewscale::new_scale_color 实现单独的线条图例，这将添加第二个颜色图例。这样做需要为您的hline 和vline 添加一个映射。

棘手的部分是默认情况下，您将获得图例键的垂直线和水平线。为了覆盖这一点，我添加了一个自定义键字形函数，该函数以 color 为条件之一。这意味着如果您更改颜色，您也必须调整条件。

library(ggplot2)

# Custom Key Glyph. Vertical Line if color = "blue". Horizontal Line otherwise
draw_key_cust <- function(data, params, size) 
  if (data$colour == "blue") 
    draw_key_vpath(data, params, size)
   else 
    draw_key_path(data, params, size)
  


ggplot(data=df,aes(index,vector)) + 
  geom_point(aes(alpha = name, color=name)) +
  scale_alpha_manual(name="Indices",
                     labels=c("legend label 1","legend label 2","legend label 3"),
                     values=c(1,0.5,0.5), guide = guide_legend(order = 1)) +
  scale_color_manual(name="Indices",
                     labels=c("legend label 1","legend label 2","legend label 3"),
                     values=c("black","orange","darkorange4"), guide = guide_legend(order = 1)) +
  ggnewscale::new_scale_color() +
  geom_vline(data = data.frame(xintercept = line1), aes(xintercept = xintercept, color="line1"), key_glyph = "cust") +
  geom_hline(data = hl, aes(yintercept = Z, color="line2"), key_glyph = "cust") +
  scale_color_manual(name="Legend 2",
                     labels=c(line1 ="line 1", line2 = "line 2"),
                     values=c(line1 = "blue", line2 = "red"), 
                     guide = guide_legend(order = 2)) +
  xlab("index") + ylab("rnorm") +  
  facet_wrap(~label_f,ncol=1,scale="free_y") +
  theme(legend.position="bottom")

【讨论】：

比我的好多了！

@langtang。谢谢。在我知道ggnewscale 之前，我的做法和你完全一样。我会说这绝对没问题。