ggplot 制作的自动裁剪多面图

autocrop faceted plots made by ggplot

在ggplot中制作多面图并更改纵横比时,通常会在图形的左右或上方和下方出现很多白色space。例如:

library(ggplot2)
df <- data.frame(x=rep(1,3), y=rep(1,3), z=factor(letters[1:3]))
p <- ggplot(df, aes(x, y)) + geom_point() + coord_fixed(ratio=1) + facet_grid(z ~ .)
ggsave("plot.jpg", p, scale=1, device="jpeg")

有没有办法自动裁剪图表?

这是我想到的。我在你的样品上测试过它,它在那里似乎工作正常,但如果它在其他地方坏了,请提前道歉。 width/height 的信息我不得不深入挖掘 grob 版本,根据剧情是否分面,"unit" 的属性信息位于不同的地方。

希望更精通 grid 包的 unit 对象的人可以参与进来,但这是我得到的:

# Note that you need to set an upper limit to the maximum height/width
# that the plot can occupy, in the max.dimension parameter (defaults to
# 10 inches in this function)

ggsave_autosize <- function(filename, plot = last_plot(), device = NULL, path = NULL, scale = 1, 
                            max.dimension = 10, units = c("in", "cm", "mm"), 
                            dpi=300, limitsize = TRUE){

  sumUnitNull <- function(x){
    res <- 0
    for(i in 1:length(x)){ 
      check.unit <- ifelse(!is.null(attr(x[i], "unit")), attr(x[i], "unit"), 
                           ifelse(!is.null(attr(x[i][[1]], "unit")), attr(x[i][[1]], "unit"), NA))
      if(!is.na(check.unit) && check.unit == "null") res <- res + as.numeric(x[i])
    }
    return(res)
  }

  # get width/height information from the plot object (likely in a mixture of different units)
  w <- ggplotGrob(plot)$widths
  h <- ggplotGrob(plot)$heights

  # define maximum dimensions
  w.max <- grid::unit(max.dimension, units) %>% grid::convertUnit("in") %>% as.numeric()
  h.max <- grid::unit(max.dimension, units) %>% grid::convertUnit("in") %>% as.numeric()

  # sum the inflexible size components of the plot object's width/height 
  # these components have unit = "in", "mm", "pt", "grobheight", etc
  w.in <- w %>% grid::convertUnit("in") %>% as.numeric() %>% sum()
  h.in <- h %>% grid::convertUnit("in") %>% as.numeric() %>% sum()

  # obtain the amount of space available for the flexible size components
  w.avail <- w.max - w.in
  h.avail <- h.max - h.in

  # sum the flexible sized components of the plot object's width/height 
  # these components have unit = "null"
  w.f <- sumUnitNull(w)
  h.f <- sumUnitNull(h)

  # shrink the amount of avilable space based on what the flexible components would actually take up
  if(w.f/h.f > w.avail/h.avail) h.avail <- w.avail/w.f*h.f else w.avail <- h.avail/h.f*w.f

  w <- w.in + w.avail
  h <- h.in + h.avail

  ggsave(filename, plot = plot, device = device, path = path, scale = scale,
         width = w, height = h, units = units, dpi = dpi, limitsize = limitsize)
}


p <- ggplot(mpg, aes(displ, cty)) + geom_point() + coord_fixed(ratio=1)
p <- p + facet_grid(. ~ cyl)

ggsave("pOriginal.png", p + ggtitle("original"))
ggsave_autosize("pAutoSize.png", p + ggtitle("auto-resize"))
ggsave_autosize("pAutoSize8.png", p + ggtitle("auto-resize, max dim = 8in x 8in"), max.dimension = 8, units = "in")

原版w/o裁剪。左边/右边有黑色space:

自动裁剪版本。高度 = 10 英寸:

自动裁剪版本。高度 = 8 英寸(所以字体看起来稍微大一点):