在ggplot2中保存每个修改过的facet

Question

我尝试使用Species将iris数据集中的每个for loop数据保存到.png文件中。但之前，我想在我的实际数据绘图过程中修改小平面条的厚度。

然而，当我试图写每个方面时 下面的代码只是给我每个物种的空图。

这是我的尝试，

library(ggplot2)
plot_list = list()
for (i in unique(iris$Species)) {
  p = ggplot(iris[iris$Species == i, ], aes(x=Sepal.Length, y=Sepal.Width)) +
    geom_point(size=3, aes(colour=Species))+
    facet_wrap(~Species)

#this part to modify facet_wrap strips
  g1 = ggplotGrob(p)

  pos =  c(unique(subset(g1$layout, grepl("panel", g1$layout$name), select = t)))
  for(i in pos) g1$heights[i-1] = unit(0.4,"cm")

  grobs = which(grepl("strip", g1$layout$name))
  for(i in grobs) g1$grobs[[i]]$heights <-  unit(1, "npc") 

  grid.newpage()
  grid.draw(g1)
  plot_list[[i]] = g1
}

#finally write the modified graphs to file


for (i in 1:3) {
  file_name = paste("iris_plot_", i, ".png", sep="")
  tiff(file_name)
  print(plot_list[[i]])
  dev.off()
}

目前这段代码正在生成空图，不知道为什么！任何帮助将不胜感激！

Answer 1

您不需要使用ggplotGrob修改条带高度。在ggplot＆＃39; s theme()中设置相关参数可以：

p1 = ggplot(iris[iris$Species == "setosa",],
            aes(x = Sepal.Length, y = Sepal.Width)) +
  geom_point() +
  facet_wrap(~Species)

p2 = p1 + theme(strip.text.x = element_text(margin = margin(t = 10, b = 10)))
# note: default margin for top & bottom is 5.5

gridExtra::grid.arrange(p1, p2, ncol = 2)

至于其余部分，您可能希望在第一次循环后检查plot_list的长度。您最初分配i以获取iris$Species的唯一值，然后尝试将其用作图表列表的索引。 plot_list的前三个元素不包含图表。

以下内容适用于此示例。您可能需要对实际用例进行一些修改：

plot_list = list()
loop.list <- unique(iris$Species)
for (i in seq_along(loop.list)) {
  p = ggplot(iris[iris$Species == loop.list[i], ], 
             aes(x = Sepal.Length, y=Sepal.Width)) +
    geom_point(size = 3, aes(colour = Species))+
    facet_wrap(~Species) +
    theme(strip.text.x = element_text(margin = margin(t = 11, b = 11)))

  plot_list[[i]] <- ggplotGrob(p)
}

for (i in 1:3) {
  file_name = paste("iris_plot_", i, ".png", sep="")
  tiff(file_name)
  grid.draw(plot_list[[i]])
  dev.off()
}