Добавить метки "плавающей оси" в графике facet_wrap

У меня та же проблема, что и этот пользователь. У меня есть "зубчатый" граненый сюжет, в котором нижняя строка имеет меньше панелей, чем другие строки, а я хотелось бы, чтобы в нижней части каждого столбца были отметки оси х.

Предлагаемое решение этой проблемы состояло в том, чтобы установить scales="free_x". (В ggplot 0.9.2.1, я считаю, что поведение, которое я искал, было дефолтом в более ранних версиях.) Это плохое решение в моем случае: мои метки фактической оси будут довольно длинными, поэтому помещать их под каждую строку будет слишком много комната. Результаты выглядят примерно так:

 x <- gl(3, 1, 15, labels=paste("this is a very long axis label ", letters[1:5]))
 y <- rnorm(length(x))
 l <- gl(5, 3, 15)
 d <- data.frame(x=x, y=y, l=l)
 ggplot(d, aes(x=x, y=y)) + geom_point() + facet_wrap(~l, scales="free_x") + 
   theme(axis.text.x=element_text(angle=90, hjust=1))

В комментарии здесь, Андри предлагает, что это можно сделать вручную в grid, но я понятия не имею, как начать с этого.

Ответ 1

Если я правильно помню, возникли вопросы о том, как добавить все метки в одну строку под последним столбцом и как поднять эти последние ярлыки до следующей строки. Итак, вот функция для обоих случаев:

Изменить:, поскольку это похоже на замену print.ggplot (см. getAnywhere(print.ggplot)). Я добавил некоторые строки для сохранения функциональности.

Изменить 2: Я немного улучшил его: больше не нужно указывать nrow и ncol, также можно печатать графики со всеми панелями.

library(grid)
# pos - where to add new labels
# newpage, vp - see ?print.ggplot
facetAdjust <- function(x, pos = c("up", "down"), 
                        newpage = is.null(vp), vp = NULL)
{
  # part of print.ggplot
  ggplot2:::set_last_plot(x)
  if(newpage)
    grid.newpage()
  pos <- match.arg(pos)
  p <- ggplot_build(x)
  gtable <- ggplot_gtable(p)
  # finding dimensions
  dims <- apply(p$panel$layout[2:3], 2, max)
  nrow <- dims[1]
  ncol <- dims[2]
  # number of panels in the plot
  panels <- sum(grepl("panel", names(gtable$grobs)))
  space <- ncol * nrow
  # missing panels
  n <- space - panels
  # checking whether modifications are needed
  if(panels != space){
    # indices of panels to fix
    idx <- (space - ncol - n + 1):(space - ncol)
    # copying x-axis of the last existing panel to the chosen panels 
    # in the row above
    gtable$grobs[paste0("axis_b",idx)] <- list(gtable$grobs[[paste0("axis_b",panels)]])
    if(pos == "down"){
      # if pos == down then shifting labels down to the same level as 
      # the x-axis of last panel
      rows <- grep(paste0("axis_b\\-[", idx[1], "-", idx[n], "]"), 
                   gtable$layout$name)
      lastAxis <- grep(paste0("axis_b\\-", panels), gtable$layout$name)
      gtable$layout[rows, c("t","b")] <- gtable$layout[lastAxis, c("t")]
    }
  }
  # again part of print.ggplot, plotting adjusted version
  if(is.null(vp)){
    grid.draw(gtable)
  }
  else{
    if (is.character(vp)) 
      seekViewport(vp)
    else pushViewport(vp)
    grid.draw(gtable)
    upViewport()
  }
  invisible(p)
}

И вот как выглядит

d <- ggplot(diamonds, aes(carat, price, fill = ..density..)) +
  xlim(0, 2) + stat_binhex(na.rm = TRUE) + theme(aspect.ratio = 1) + 
  facet_wrap(~ color)
facetAdjust(d)

facetAdjust(d, "down")

Изменить 3:

Это альтернативное решение, одно из которых отлично.

Есть некоторые проблемы, когда вы хотите использовать ggsave вместе с facetAdjust. График класса ggplot требуется из-за двух частей в исходном коде ggsave: print(plot) и default_name(plot) в случае, если вы не укажете имя файла вручную (согласно ?ggsave кажется, что это не предполагается работать, хотя). Следовательно, с учетом имени файла существует обходное решение (возможно, с некоторыми побочными эффектами):

Сначала рассмотрим отдельную функцию, которая достигает основного эффекта плавающей оси. Обычно он возвращает объект gtable, однако мы используем class(gtable) <- c("facetAdjust", "gtable", "ggplot"). Таким образом, разрешено использовать ggsave и print(plot) работает по мере необходимости (см. Ниже для print.facetAdjust)

facetAdjust <- function(x, pos = c("up", "down"))
{
  pos <- match.arg(pos)
  p <- ggplot_build(x)
  gtable <- ggplot_gtable(p); dev.off()
  dims <- apply(p$panel$layout[2:3], 2, max)
  nrow <- dims[1]
  ncol <- dims[2]
  panels <- sum(grepl("panel", names(gtable$grobs)))
  space <- ncol * nrow
  n <- space - panels
  if(panels != space){
    idx <- (space - ncol - n + 1):(space - ncol)
    gtable$grobs[paste0("axis_b",idx)] <- list(gtable$grobs[[paste0("axis_b",panels)]])
    if(pos == "down"){
      rows <- grep(paste0("axis_b\\-[", idx[1], "-", idx[n], "]"), 
                   gtable$layout$name)
      lastAxis <- grep(paste0("axis_b\\-", panels), gtable$layout$name)
      gtable$layout[rows, c("t","b")] <- gtable$layout[lastAxis, c("t")]
    }
  }
  class(gtable) <- c("facetAdjust", "gtable", "ggplot"); gtable
}

Функция для печати, которая отличается только несколькими строками от ggplot2:::print.ggplot:

print.facetAdjust <- function(x, newpage = is.null(vp), vp = NULL) {
  if(newpage)
    grid.newpage()
  if(is.null(vp)){
    grid.draw(x)
  } else {
    if (is.character(vp)) 
      seekViewport(vp)
    else pushViewport(vp)
    grid.draw(x)
    upViewport()
  }
  invisible(x)
}

Пример:

d <- ggplot(diamonds, aes(carat, price, fill = ..density..)) +
  xlim(0, 2) + stat_binhex(na.rm = TRUE) + theme(aspect.ratio = 1) + 
  facet_wrap(~ color)
p <- facetAdjust(d) # No output
print(p) # The same output as with the old version of facetAdjust()
ggsave("name.pdf", p) # Works, a filename is necessary