为循环中创建的ggplot列表添加图例

时间:2017-07-19 09:05:47

标签: r loops ggplot2 legend

我正在使用ggplot2中的循环创建一个绘图列表(没有图例)。然后我分别创建了一个图例,并尝试使用grid.arrange和grobs函数打印组合图。它创建了组合图,但没有图例。有人可以帮忙解决问题吗?

我在这里附上我的代码:

df1<-data.frame(x=1:10,y1=rnorm(10),y2=rnorm(10),y3=rnorm(10),y4=rnorm(10),y5=rnorm(10))
df2 <- melt(df1,id.vars="x")

plot.list = list()

for (i in 1:3){
  p <- ggplot(df2, aes(x=x, y=value)) + 
  geom_line(aes(colour=variable, group=variable))+
    theme(legend.position='none')
  plot.list[[i]] = p
}

temp_legend <- ggplot(df2, aes(x=x, y=value)) + 
  geom_line(aes(colour=variable, group=variable)) + 
  scale_color_manual("",labels = c("Observed","3d","5d","7d","10d"), values = c("black", "limegreen","blue","tan1","red3")) +
  theme(legend.direction = "horizontal", 
        legend.position = "bottom")

library(gridExtra)
# create get_legend function
get_legend<-function(myggplot){
  tmp <- ggplot_gtable(ggplot_build(myggplot))
  leg <- which(sapply(tmp$grobs, function(x) x$name) == "guide-box")
  legend <- tmp$grobs[[leg]]
  return(legend)
}

# Extract legend using get_legend function
legend <- get_legend(temp_legend)

# png export
png("Output.png", width = 5, height = 6.35, units = 'in', res = 300)
grid.arrange(grobs=plot.list,legend, 
             ncol=1, nrow = 4, 
             widths = 6, heights = c(2,2,2,0.35))
dev.off()

1 个答案:

答案 0 :(得分:2)

.Rprofile

或只是

grid.arrange(grobs=c(plot.list,list(legend)), 
             ncol=1, heights = c(2,2,2,0.35))