使用arrangeGrob将子图添加为图例

时间:2016-07-25 04:18:17

标签: r ggplot2 gridextra grob

我正在尝试将两个地块绘制在一起,其中一个是主要情节,另一个是我希望位于主要情节的图例区域的子情节。

此代码生成此图:

gLegend <- function(a.gplot){
  tmp <- ggplot_gtable(ggplot_build(a.gplot))
  leg <- which(sapply(tmp$grobs, function(x) x$name) == "guide-box")
  legend <- tmp$grobs[[leg]]
  return(legend)
} 

set.seed(1)
main.df <- data.frame(group=rep(LETTERS[1:4],3),
                     y=rnorm(12),x=c(rep(1,4),rep(2,4),rep(3,4)),col=rep(c("gray","blue","red","magenta"),3))
main.df$group <- factor(main.df$group,levels=LETTERS[1:4])
sub.df <- data.frame(group=c("B","C","D"),x=1:3,effect=runif(3,0,1),col=c("blue","red","magenta"))
sub.df$group <- factor(sub.df$group,levels=LETTERS[2:4])

main.plot <- ggplot(main.df,aes(x=x,y=y,color=factor(group)))+geom_point(size=3)+facet_wrap(~group,ncol=4)+scale_fill_manual(values=c("gray","blue","red","magenta"),labels=c("A","B","C","D"),name="group")+scale_colour_manual(values=c("gray","blue","red","magenta"),labels=c("A","B","C","D"),name="group")+scale_x_continuous(breaks=unique(main.df$x))
sub.plot <- ggplot(sub.df,aes(x=x,y=effect,color=factor(group)))+geom_point(size=2)+scale_fill_manual(values=c("blue","red","magenta"),labels=c("B","C","D"),name="group",guide=FALSE)+scale_colour_manual(values=c("blue","red","magenta"),labels=c("B","C","D"),name="group",guide=FALSE)+labs(x="group",y="effect")+ggtitle("effect summary")+scale_x_continuous(breaks=unique(sub.df$x),labels=c("B","C","D"))

sub.plot.grob <- ggplotGrob(sub.plot)
combined.plot <- arrangeGrob(main.plot+theme(legend.position="none"),widths=c(0.75,0.25),arrangeGrob(20,sub.plot.grob),ncol=2)

enter image description here

我想做的是摆脱主要情节的传说。

如果我将guide=FALSE添加到scale_fill_manual scale_colour_manual gLegend会引发错误:

Error in tmp$grobs[[leg]] : 
  attempt to select less than one element in get1index

显然是因为它不能很好"guide-box"

知道如何解决这个问题吗?

3 个答案:

答案 0 :(得分:1)

enter image description here以下是使用cowplot

的替代解决方案
library(ggplot2)
library(cowplot)
theme_set(theme_bw()) #cowplot automatically sets theme_classic

set.seed(1)
main.df <- data.frame(group=rep(LETTERS[1:4],3),
                      y=rnorm(12),x=c(rep(1,4),rep(2,4),rep(3,4)),col=rep(c("gray","blue","red","magenta"),3))
main.df$group <- factor(main.df$group,levels=LETTERS[1:4])
sub.df <-     data.frame(group=c("B","C","D"),x=1:3,effect=runif(3,0,1),col=c("blue","red","magenta"))
sub.df$group <- factor(sub.df$group,levels=LETTERS[2:4])

sub.plot <- ggplot(sub.df,aes(x=x,y=effect,color=factor(group)))+
  geom_point(size=2)+
  scale_fill_manual(values=c("blue","red","magenta"),labels=c("B","C","D"),name="group",guide=FALSE)+
  scale_colour_manual(values=c("blue","red","magenta"),labels=c("B","C","D"),name="group",guide=FALSE)+
  labs(x="group",y="effect")+
  ggtitle("effect summary")+
  scale_x_continuous(breaks=unique(sub.df$x),labels=c("B","C","D"))


main.plot <- ggplot(main.df,aes(x=x,y=y,color=factor(group)))+
  geom_point(size=3)+facet_wrap(~group,ncol=4)+
  scale_fill_manual(values=c("gray","blue","red","magenta"),labels=c("A","B","C","D"),name="group")+
  scale_colour_manual(values=c("gray","blue","red","magenta"),labels=c("A","B","C","D"),name="group")+
  scale_x_continuous(breaks=unique(main.df$x)) +
  theme(legend.position = "none")

ggdraw() +
  draw_plot(main.plot, x = 0, y = 0, width = 0.75, height = 1) +
  draw_plot(sub.plot, 0.75, 0, .25, .5)

答案 1 :(得分:1)

试试这个,

grid.arrange(ggplot(), ggplot(), 
             layout_matrix=matrix(c(1,1,NA,2), 2), widths=c(1.5,1))

enter image description here

答案 2 :(得分:0)

以下是使用grid包的解决方案,如果您想要一个空格,其中是当前图例:

enter image description here

library(grid)

set.seed(1)
main.df <- data.frame(group=rep(LETTERS[1:4],3),
                  y = rnorm(12),
                  x = c(rep(1,4), rep(2,4), rep(3,4)),
                  col = rep(c("gray", "blue", "red", "magenta"), 3))
main.df$group <- factor(main.df$group,levels=LETTERS[1:4])

sub.df <- data.frame(group=c("B", "C", "D"),
                 x = 1:3,
                 effect = runif(3, 0, 1),
                 col = c("blue", "red", "magenta"))
sub.df$group <- factor(sub.df$group, levels=LETTERS[2:4])

main.plot <- ggplot(main.df, aes(x = x,y = y, color = factor(group))) + 
geom_point(size = 3) + facet_wrap(~group, ncol = 4) + 
scale_fill_manual(values = c("gray", "blue", "red", "magenta"), 
labels = c("A", "B", "C", "D"),name = "group") + 
scale_colour_manual(values = c("gray", "blue", "red", "magenta"), 
labels = c("A", "B", "C", "D"), name = "group") + 
scale_x_continuous(breaks = unique(main.df$x)) + guides(color = F)

sub.plot <- ggplot(sub.df, aes(x = x, y = effect, color = factor(group))) +
geom_point(size = 2) + 
scale_fill_manual(values = c("blue", "red", "magenta"), 
labels = c("B", "C", "D"), name= "group" ,guide = FALSE) + 
scale_colour_manual(values = c("blue", "red", "magenta"), 
labels = c("B", "C", "D"), name = "group", guide=FALSE) + 
labs(x = "group", y = "effect") + 
ggtitle("effect summary") + 
scale_x_continuous(breaks = unique(sub.df$x), labels = c("B", "C", "D"))


blank = grid.rect(gp = gpar(col = "white"))

sub.plot.grob <- arrangeGrob(blank, sub.plot, ncol = 1)

combined.plot <- arrangeGrob(main.plot, sub.plot.grob, ncol=2, widths=c(0.75,0.25))

grid.arrange(combined.plot)