反ggplotGrob?

时间:2014-10-22 02:54:42

标签: r ggplot2 grob

我有一个操作ggplot对象的函数,通过将其转换为grob然后修改图层。我希望函数返回一个ggplot对象而不是grob。有没有一种简单的方法将grob转换回gg?

ggplotGrob上的{p> The documentation非常稀疏 简单的例子:

P <- ggplot(iris) + geom_bar(aes(x=Species, y=Petal.Width), stat="identity")

G <- ggplotGrob(P)
... some manipulation to G ...

## DESIRED: 
P2 <- inverse_of_ggplotGrob(G)

such that, we can continue to use basic ggplot syntax, ie
`P2 + ylab ("The Width of the Petal")`

更新:

要回答评论中的问题,这里的动机是根据每个方面中标签名称的值,以编程方式修改构面标签的颜色。以下功能很好地工作(基于前一个问题中的baptise输入)。

我希望colorByGroup的返回值是一个ggplot对象,而不仅仅是一个grob。

以下是感兴趣的人的代码

get_grob_strips <- function(G, strips=grep(pattern="strip.*", G$layout$name)) {

  if (inherits(G, "gg"))
    G <- ggplotGrob(G)
  if (!inherits(G, "gtable"))
    stop ("G must be a gtable object or a gg object")

  strip.type <- G$layout[strips, "name"]
  ## I know this works for a simple 
  strip.nms <- sapply(strips, function(i) {
     attributes(G$grobs[[i]]$width$arg1)$data[[1]][["label"]]
  })

  data.table(grob_index=strips, type=strip.type, group=strip.nms)
}


refill <- function(strip, colour){
  strip[["children"]][[1]][["gp"]][["fill"]] <- colour
  return(strip)
}

colorByGroup <- function(P, colors, showWarnings=TRUE) {
## The names of colors should match to the groups in facet
  G <- ggplotGrob(P)
  DT.strips <- get_grob_strips(G)

  groups <- names(colors)
  if (is.null(groups) || !is.character(groups)) {
    groups <- unique(DT.strips$group)
    if (length(colors) < length(groups))
      stop ("not enough colors specified")
    colors <- colors[seq(groups)]
    names(colors) <- groups
  }


  ## 'groups' should match the 'group' in DT.strips, which came from the facet_name
  matched_groups <- intersect(groups, DT.strips$group)
  if (!length(matched_groups))
    stop ("no groups match")
  if (showWarnings) {
      if (length(wh <- setdiff(groups, DT.strips$group)))
        warning ("values in 'groups' but not a facet label: \n", paste(wh, colapse=", "))
      if (length(wh <- setdiff(DT.strips$group, groups)))
        warning ("values in facet label but not in 'groups': \n", paste(wh, colapse=", "))
  }

  ## identify the indecies to the grob and the appropriate color
  DT.strips[, color := colors[group]]
  inds <- DT.strips[!is.na(color), grob_index]
  cols <- DT.strips[!is.na(color), color]

  ## Fill in the appropriate colors, using refill()
  G$grobs[inds] <- mapply(refill, strip = G$grobs[inds], colour = cols, SIMPLIFY = FALSE)

  G
}

3 个答案:

答案 0 :(得分:10)

我会说不。 ggplotGrob是一条单行道。 grob对象是绘制由网格定义的基元。您可以从头开始创建任意grob。没有通用的方法可以将随机的grobs集合转换回生成它们的函数(它不可逆,因为它不是1:1)。一旦你去了grob,你永远不会回去。

您可以将ggplot对象包装在自定义类中,并重载plot / print命令以执行一些自定义grob操作,但这可能更糟糕。

答案 1 :(得分:4)

您可以尝试以下操作:

p = ggplotify::as.ggplot(g)

有关更多信息,请参见https://cran.r-project.org/web/packages/ggplotify/vignettes/ggplotify.html

它涉及一些作弊annotation_custom(as.grob(plot),...),因此它可能无法在所有情况下都有效:https://github.com/GuangchuangYu/ggplotify/blob/master/R/as-ggplot.R

答案 2 :(得分:3)

看看ggpubr软件包:它具有函数as_ggplot()。如果您的杂技不太复杂,则可能是一个解决方案!

我还建议您看看结合了ggplots的patchwork package……可能不是您要找的东西,而是……看看。