Как я могу сделать графики согласованной ширины в ggplot (с легендами)?

У меня есть несколько разных категорий, которые я хочу построить. Это разные категории, каждый со своим набором ярлыков, но имеет смысл группироваться вместе в документе. Ниже приведены некоторые простые примеры столбчатых диаграмм:

df <- data.frame(x=c("a", "b", "c"),
 y=c("happy", "sad", "ambivalent about life"))
ggplot(df, aes(x=factor(0), fill=x)) + geom_bar()
ggplot(df, aes(x=factor(0), fill=y)) + geom_bar()

Проблема в том, что с разными ярлыками легенды имеют разную ширину, что означает, что графики имеют разную ширину, что приводит к тому, что вещи выглядят немного тупыми, если я создаю таблицы или элементы \subfigure. Как я могу это исправить?

Есть ли способ явно установить ширину (абсолютную или относительную) как графика, так и легенды?

4 ответа

Изменить: Очень легко с пакетом egg, доступным в github

# install.package(devtools)
# devtools::install_github("baptiste/egg")
library(egg)
p1 <- ggplot(data.frame(x=c("a","b","c"),
 y=c("happy","sad","ambivalent about life")),
 aes(x=factor(0),fill=x)) + 
 geom_bar()
p2 <- ggplot(data.frame(x=c("a","b","c"),
 y=c("happy","sad","ambivalent about life")),
 aes(x=factor(0),fill=y)) + 
 geom_bar()
ggarrange(p1,p2, ncol = 1)

Оригинал Устаревший к ggplot2 2.2.1

Здесь используется решение, которое использует функции из пакета gtable и фокусируется на ширинах полей легенды. (Более общее решение можно найти здесь.)

library(ggplot2) 
library(gtable) 
library(grid)
library(gridExtra) 
# Your plots
p1 <- ggplot(data.frame(x=c("a","b","c"),y=c("happy","sad","ambivalent about life")),aes(x=factor(0),fill=x)) + geom_bar()
p2 <- ggplot(data.frame(x=c("a","b","c"),y=c("happy","sad","ambivalent about life")),aes(x=factor(0),fill=y)) + geom_bar()
# Get the gtables
gA <- ggplotGrob(p1)
gB <- ggplotGrob(p2)
# Set the widths
gA$widths <- gB$widths
# Arrange the two charts.
# The legend boxes are centered
grid.newpage()
grid.arrange(gA, gB, nrow = 2)

Если, кроме того, поля легенды необходимо оставить обоснованными и заимствовать код здесь, написанный @Julius

p1 <- ggplot(data.frame(x=c("a","b","c"),y=c("happy","sad","ambivalent about life")),aes(x=factor(0),fill=x)) + geom_bar()
p2 <- ggplot(data.frame(x=c("a","b","c"),y=c("happy","sad","ambivalent about life")),aes(x=factor(0),fill=y)) + geom_bar()
# Get the widths
gA <- ggplotGrob(p1)
gB <- ggplotGrob(p2)
# The parts that differs in width
leg1 <- convertX(sum(with(gA$grobs[[15]], grobs[[1]]$widths)), "mm")
leg2 <- convertX(sum(with(gB$grobs[[15]], grobs[[1]]$widths)), "mm")
# Set the widths
gA$widths <- gB$widths
# Add an empty column of "abs(diff(widths)) mm" width on the right of 
# legend box for gA (the smaller legend box)
gA$grobs[[15]] <- gtable_add_cols(gA$grobs[[15]], unit(abs(diff(c(leg1, leg2))), "mm"))
# Arrange the two charts
grid.newpage()
grid.arrange(gA, gB, nrow = 2)

Альтернативные решения В пакете gtable есть функции rbind и cbind для объединения grobs в один grob. Для диаграмм здесь ширины должны быть установлены с помощью size = "max", но версия CRAN gtable выдает ошибку.

Один вариант: должно быть очевидно, что легенда на втором графике шире. Поэтому используйте опцию size = "last".

# Get the grobs
gA <- ggplotGrob(p1)
gB <- ggplotGrob(p2)
# Combine the plots
g = rbind(gA, gB, size = "last")
# Draw it
grid.newpage()
grid.draw(g)

Левые легенды:

# Get the grobs
gA <- ggplotGrob(p1)
gB <- ggplotGrob(p2)
# The parts that differs in width
leg1 <- convertX(sum(with(gA$grobs[[15]], grobs[[1]]$widths)), "mm")
leg2 <- convertX(sum(with(gB$grobs[[15]], grobs[[1]]$widths)), "mm")
# Add an empty column of "abs(diff(widths)) mm" width on the right of 
# legend box for gA (the smaller legend box)
gA$grobs[[15]] <- gtable_add_cols(gA$grobs[[15]], unit(abs(diff(c(leg1, leg2))), "mm"))
# Combine the plots
g = rbind(gA, gB, size = "last")
# Draw it
grid.newpage()
grid.draw(g)

Второй вариант - использовать rbind из пакета Baptiste gridExtra

# Get the grobs
gA <- ggplotGrob(p1)
gB <- ggplotGrob(p2)
# Combine the plots
g = gridExtra::rbind.gtable(gA, gB, size = "max")
# Draw it
grid.newpage()
grid.draw(g)

Левые легенды:

# Get the grobs
gA <- ggplotGrob(p1)
gB <- ggplotGrob(p2)
# The parts that differs in width
leg1 <- convertX(sum(with(gA$grobs[[15]], grobs[[1]]$widths)), "mm")
leg2 <- convertX(sum(with(gB$grobs[[15]], grobs[[1]]$widths)), "mm")
# Add an empty column of "abs(diff(widths)) mm" width on the right of 
# legend box for gA (the smaller legend box)
gA$grobs[[15]] <- gtable_add_cols(gA$grobs[[15]], unit(abs(diff(c(leg1, leg2))), "mm"))
# Combine the plots
g = gridExtra::rbind.gtable(gA, gB, size = "max")
# Draw it
grid.newpage()
grid.draw(g)


Как указывает @hadley, rbind.gtable должен иметь возможность справиться с этим,

grid.draw(rbind(ggplotGrob(p1), ggplotGrob(p2), size="last"))

однако, в идеале ширина макета должна быть size="max", которая не справляется хорошо с некоторыми типами узлов сетки.


Пакет cowplot также имеет функцию align_plots для этой цели (вывод не показан),

both2 <- align_plots(p1, p2, align="hv", axis="tblr")
p1x <- ggdraw(both2[[1]])
p2x <- ggdraw(both2[[2]])
save_plot("cow1.png", p1x)
save_plot("cow2.png", p2x)

а также plot_grid, который сохраняет графики в один и тот же файл.

library(cowplot)
both <- plot_grid(p1, p2, ncol=1, labels = c("A", "B"), align = "v")
save_plot("cow.png", both)


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

Арун предложил переместить легенду сверху или снизу:

ggplot(df, aes(x=factor(0), fill=x)) + geom_bar() + theme(legend.position = "bottom")
ggplot(df, aes(x=factor(0), fill=y)) + geom_bar() + theme(legend.position = "bottom")

Теперь графики имеют ту же ширину, что и запрос. Кроме того, площадь участка одинакового размера в обоих случаях.

Если есть больше факторов или даже более длинных меток, может возникнуть необходимость играть с легендой, например, для отображения легенды в двух или более строках. theme() и guide_legend() имеют несколько параметров для управления положением и появлением легенд в ggplot2.

licensed under cc by-sa 3.0 with attribution.