图例没有使用ggplot2绘制多个密度图

时间:2016-08-09 16:50:46

标签: r plot ggplot2 legend

正如标题所说,下面的函数从传递的数据中绘制了许多密度,但没有绘制图例。 MWE:

plotDensities <- function(xlab="", xlim=c(), ...) {
  datas <- list(...)
  cbPalette <- c("#000000", "#E69F00", "#56B4E9", "#009E73", "#F0E442", "#0072B2", "#D55E00", "#CC79A7")
  if (length(datas) > length(cbPalette)) return(invisible(NULL))
  dplot <- ggplot() + xlab(xlab) + 
    scale_colour_manual(values=setNames(cbPalette[1:length(datas_names)], datas_names)) +
    theme(legend.position = c(.8, .8))
  datas_names <- names(datas)
  for (i in 1:length(datas)) {
    name <- datas_names[i]
    values <- data.frame(x=datas[[name]])
    dplot <- dplot + geom_density(aes(x=x), colour=cbPalette[i], data=values)
  }
  if (!is.null(xlim))
    dplot <- dplot + xlim(xlim)

  return(invisible(dplot))
}
v1 <- rnorm(2000, 0, 1)
v2 <- rnorm(3000, 1, 1.5)
v3 <- rnorm(4000, 2, 2.5)
dplot <- plotDensities(xlim="whatever", v1=v1, v2=v2, v3=v3)
dplot

1 个答案:

答案 0 :(得分:1)

将数据中的列映射到ggplot内的美学(例如colourfill等)时,

aes会生成图例。但是,您的代码在colour的{​​{1}}调用之外设置了aes,而geom_density则无法生成图例。

此外,您的数据被分成不同的向量。但是,如果您在&#34; long&#34;中有一个数据框,则ggplot效果最佳。格式。在这种情况下,这意味着将v1v2v3合并到一个数据框中,另一列标记每行最初来自哪个矢量。

这是一个简单的例子,使用您的数据,您可以适应您的功能:

# Convert data to long format
dat = data.frame(vals=c(v1,v2,v3), 
                 source=rep(c("v1","v2","v3"),sapply(list(v1,v2,v3), length))) 

dat
           vals source
1    -0.2860619     v1
2     1.8072476     v1
3    -0.6492385     v1
...
8998  3.8989335     v3
8999  5.1929588     v3
9000  1.8716558     v3
ggplot(dat, aes(vals, colour=source)) +
  geom_density() +
  scale_colour_manual(values=cbPalette[1:length(unique(dat$source))])

enter image description here