根据构面的存在有条件地修改 ggplot 主题？

我正在开发一个自定义 ggplot2 主题，并认为根据绘图对象的某些特征自动修改主题元素可能会很漂亮。例如，是否有一种方法可以指定如果绘图包含小平面，则为每个面板添加边框？

我想问题确实是，我可以从自定义 theme() 调用中访问当前的 gg 对象，然后有条件地应用某些主题元素吗？在我的脑海中，我将主题函数定义为如下所示：

theme_custom <- function() {
  if (plot$facet$params > 0) {
  theme_minimal() +
    theme(panel.border = element_rect(color = "gray 50", fill = NA))
  }
  else {
    theme_minimal()
    }
}

如果这是可能的，那么在使用中它会看起来像这样：

library(ggplot2)

# plot with facets automatically adds panel borders
ggplot(mtcars, aes(mpg, wt)) +
  geom_point() +
  facet_wrap(vars(cyl)) +
  theme_custom()

# plot without facets no panel border
ggplot(mtcars, aes(mpg, wt)) +
  geom_point() +
  theme_custom() 

注意：这最初发布于RStudio 社区并没有收到答复。

我认为奥利弗的想法是正确的。

我不认为theme_customfunction 是检查条件主题绘图的正确位置，因为主题函数大多不知道它们所添加到的精确绘图。

相反，我认为检查情节的适当位置是在将主题添加到情节中时。您可以编写如下所示的主题函数，该函数为输出设置不同的类。

theme_custom <- function() {
  out <- theme_minimal()
  class(out) <- c("conditional_theme", class(out))
  out
}

现在，每次将主题添加到情节中时，都是通过ggplot_add.theme函数，我们可以将其重写为conditional_theme班级。在我看来，检查情节是否分面的正确方法是检查情节的类别plot$facet槽，可以是FacetGrid, FacetWrap等当添加适当的方面并默认为FacetNull当没有设置facet时。

ggplot_add.conditional_theme <- function(object, plot, object_name) {
  if (!inherits(plot$facet, "FacetNull")) {
    object <- object + theme(panel.border = element_rect(colour = "grey50", fill = NA))
  }
  plot$theme <- ggplot2:::add_theme(plot$theme, object, object_name)
  plot
}

现在用例应该按预期工作：

ggplot(mtcars, aes(mpg, wt)) +
  geom_point() +
  facet_wrap(vars(cyl)) +
  theme_custom()

ggplot(mtcars, aes(mpg, wt)) +
  geom_point() +
  theme_custom() 

唯一的缺点是你每次都必须将主题添加到情节中，并且不能使用theme_set(theme_custom())将此应用于会话中的任何绘图。