R:在列表中保存ggplot2图

时间:2012-07-06 06:43:27

标签: r ggplot2 histogram

我正在编写一个R代码,允许用户从数据中选择列并绘制每个列的直方图。因此,我使用'for'循环使用ggplot2库生成所需数量的绘图并将它们保存在单个列表中。但我面临的问题是,在'for'循环的每次迭代中,列表中的所有对象都存储相同的图。因此,最终输出由直方图网格组成,标记不同但描绘相同(最后)列。

我知道这个问题很陈旧,我发现renaming ggplot2 graphs in a for loophttps://stat.ethz.ch/pipermail/r-help/2008-February/154438.html上的答案是一个有用的起点。

我使用R中可用的标准瑞士生育数据集来生成图表。这是代码: -

data_ <- swiss
data_ <- na.omit(data_)

u <- c(2, 3, 4, 5, 6)
plotData <- data_[,u]
bw <- 5
plotType <- 'probability'

library(ggplot2)
library(gridExtra)

histogramList <- vector('list', length(u))

if(plotType=='probability')
{
 for(i in 1:length(u))
 {
   indexDataFrame <- data.frame(plotData[,i])
   probabilityHistogram <- ggplot(indexDataFrame, aes(x=indexDataFrame[,1]))
   histogramList[[i]] <-  probabilityHistogram + geom_histogram(aes(y=..density..),     binwidth=bw, colour='black', fill='skyblue') + geom_density() + scale_x_continuous(names(plotData)[i]) + opts(legend.position='none')
 }
} else
if(plotType=='frequency')
{
 for(i in 1:length(u))
 {
   indexDataFrame <- data.frame(plotData[,i])
   probabilityHistogram <- ggplot(indexDataFrame, aes(x=indexDataFrame[,1]))
   histogramList[[i]] <- probabilityHistogram + geom_histogram(aes(y=..count..), binwidth=bw, colour='black', fill='skyblue') + geom_density() + scale_x_continuous(names(plotData)[i]) + opts(legend.position='none')
 }
}

arg_list <- c(histogramList, list(nrow=3, ncol=2))
#jpeg('histogram', width=1024, height=968)
do.call(grid.arrange, arg_list)
#graphics.off()

如果我在本论坛中错过了对问题的明显答案,我深表歉意,如果你能指导我,我将不胜感激。我希望我的解释清楚,如果没有,请告诉我所需的澄清。

谢谢!

2 个答案:

答案 0 :(得分:17)

您可以通过以下方式大大简化代码:

  1. 使用构面,而不是手动排列多个图
  2. 使用包melt
  3. 中的reshape2功能融合您的数据
  4. 这意味着您可以删除循环
  5. 这是对代码的完全重写,看不到循环。

    data_ <- swiss
    data_ <- na.omit(data_)
    
    u <- c(2, 3, 4, 5, 6)
    plotData <- data_[,u]
    bw <- 5
    plotType <- 'frequency'
    
    library(ggplot2)
    library(reshape2)
    
    mdat <- melt(plotData)
    
    if(plotType=='probability'){
      ph <- ggplot(mdat, aes(value)) +
        geom_histogram(aes(y=..density..), binwidth=bw, colour='black', fill='skyblue') + 
        geom_density() + 
        facet_wrap(~variable, scales="free")
    } 
    
    if(plotType=='frequency'){
      ph <- ggplot(mdat, aes(value)) +
        geom_histogram(aes(y=..count..), binwidth=bw, colour='black', fill='skyblue') + 
        geom_density() + 
        facet_wrap(~variable, scales="free")
    }
    
    print(ph)
    

    生成的图形:

    <强>概率:

    enter image description here

    <强>频率

    enter image description here

答案 1 :(得分:6)

不是使用aes来映射美学,而是最好使用aes_string

 for(i in 1:length(u))
 {
   probabilityHistogram <- ggplot(plotData, aes_string(x=names(plotData)[i]))
   histogramList[[i]] <-  probabilityHistogram + geom_histogram(aes(y=..density..),     binwidth=bw, colour='black', fill='skyblue') + geom_density() + scale_x_continuous(names(plotData)[i]) + opts(legend.position='none')
 }

至少对我有用。这样可以避免必须对数据进行子集化,并允许您通过引用名称引用要绘制的列。