使用ggplot2为两个不同的geom'混合“传说'有色'和'填充'

时间:2014-09-14 20:49:32

标签: r ggplot2 legend

我想在一个单独的图和垂直线中放置三个类别的核密度图,以指示整个分布的平均值和中值。使用ggplot2可以很容易地绘制情节,但是我一直在努力使传奇变得正确并且非常努力。

set.seed(1234)
data <- data.frame(value = rgamma(n = 10000, shape = 3, scale = 1),
               type = sample(letters[1:3], size = 10000, replace = TRUE))
data$value[data$type == "b"] <- data$value[data$type == "b"] +
                          rnorm(sum(data$type == "b"), mean = 2)
data$value[data$type == "c"] <- data$value[data$type == "c"] +
                          rnorm(sum(data$type == "c"), mean = 4)

# Let's produce a 'coloured' AND 'filled' density plot
# ('cause I want both the area under the curve and the line to be coloured)
library(ggplot2)
gp <- ggplot(data=data, aes_string(x="value"))
gp <- gp + geom_density(aes_string(fill="type", colour="type"), alpha=0.3)
gp

这给了我没有问题的情节和传奇。

plot1

现在,我添加了垂直线,情节很好,但传说真的很难看。

# Now let's add vertical lines to the plot, indicating the mean 
# and median for the whole distribution
vlines <- data.frame(mean_median = c(mean(data$value), median(data$value)),
                 labels = c("Mean", "Median"))

gp <- gp + geom_vline(data=vlines,
                  aes(xintercept=mean_median, colour=labels),
                  size=1.05, linetype="dashed", show_guide=TRUE)
gp

plot2

这个传奇故事与“彩色”混合在一起。 vlines和&#39;彩色&#39;和#39;填充&#39;密度。我确实想要两个传说,其中一个是vlines的传说(两个条目),另一个想要密度图颜色的图例。

一种解决方法是让geom_density为彩色或填充,但不是两者都有。它效果更好,但它不是我想要的(因为我想要两者,密度图的面积和线条都有颜色)。它是这样的。而不是这样做:

gp <- gp + geom_density(aes_string(fill="type", colour="type"), alpha=0.3)

我这样做:

gp <- gp + geom_density(aes_string(fill="type"), alpha=0.3)

plot3

结果几乎是我想要的,但我真的希望曲线下的区域和密度图的线都是彩色的。

我一直试图在整个网络上找到解决方案,但似乎没有任何效果。这些问题有时会使用geoms的show_guide来解决,因为你可以覆盖是否为各个geoms绘制图例。我玩它并且在我的情况下不起作用,因为我需要两个geom(密度和vlines)的传说,问题是在其中一个geoms中加上填充和颜色,加上颜色其他

Stack Overflow(bar and line plot in one chart with a legend under ggplot2)中发布了一个非常类似的问题,但在那里使用的解决方案(使用子集)不适用于我的情况。

我真的很感激任何想法。我已经挣扎了很长一段时间了,我找不到解决办法。

1 个答案:

答案 0 :(得分:4)

这很可能是一种更简单的方法,但你可以破解情节对象,作为上述评论的替代方案。您可以根据自己的喜好制作两个情节,也可以根据自己的喜好制作一个情节,并交换传说。

# Plot with density area and line coloured but legend not right
p1 <- ggplot(data=data, aes(x=value)) + 
            geom_density(aes(fill=type, colour=type), alpha=0.3 ) + 
            geom_vline(data=vlines, aes(xintercept=mean_median, colour=labels), 
                         linetype="dashed", size=1.5, show_guide=TRUE ) 

g1 <- ggplotGrob(p1)

# Plot with density line not coloured but legend is ok
p2 <- ggplot(data=data, aes(x=value)) + 
            geom_density(aes(fill=type), alpha=0.3 ) + 
            geom_vline(data=vlines, aes(xintercept=mean_median, colour=labels), 
                         linetype="dashed", size=1.5, show_guide=TRUE )  +
            guides(fill = guide_legend(override.aes = list(linetype = 0 ))) 

g2 <- ggplotGrob(p2)


# Add legend of second plot to first plot    
g1$grobs[which(g1$layout$name=="guide-box")] <- 
                              g2$grobs[which(g2$layout$name=="guide-box")]  

grid::grid.newpage()    
grid::grid.draw(g1)

enter image description here