Как выровнять столбцы по вертикали, чтобы получить одинаковую ширину для каждого столбца с разными числами х - PullRequest
1 голос
/ 23 апреля 2020

Я использую пакет cowplot для создания сетки графика. Моя проблема возникает, когда я хочу построить вертикально два графика разной ширины. Вот пример:

library(dplyr)
library(ggplot2)
library(cowplot)

plot1 = iris %>% 
  ggplot(aes(x = Species, y = Sepal.Width, fill = Species)) +
  geom_col()

plot2 = iris %>% 
  filter(Species != 'virginica') %>% 
  ggplot(aes(x = Species, y = Sepal.Width, fill = Species)) +
  geom_col()

w1 = max(layer_data(plot1, 1)$x)
w2 = max(layer_data(plot2, 1)$x)

plot_grid(plot1, plot2, align = 'v', ncol = 1, rel_widths = c(w1, w2), axis = 'l')

enter image description here

Как вы можете видеть в коде, я использую функцию layer_data () для Извлеките количество столбцов в графике, потому что я хочу запустить его рекурсивно, а иногда некоторые группы удаляются, поэтому я проверяю количество столбцов. Таким образом, цель состоит в том, чтобы выровнять столбцы по вертикали на разных графиках. В предыдущем коде аргумент rel_width не действует.

Я пробовал что-то вроде этого:

plot_grid(plot1,
          plot_grid(plot2, NA, align = 'h', ncol = 2, rel_widths = c(w2, w1-w2)),
          align = 'v', ncol = 1, axis = 'lr')

Но это не работает, как ожидалось, и зависит от того, что w1 > w2. Хотелось бы получить некоторую помощь , Цель состояла бы в том, чтобы выровнять ось X от обоих графиков. Выравнивание легенды не понадобится, только ось X.

library(ggplot2)
library(cowplot)

d1 = data.frame(length = c('large', 'medium', 'small'),
                meters = c(100, 50, 30))

d2 = data.frame(speed = c('high', 'slow'),
                value =c(200, 45))

p1 = ggplot(d1, aes(x = length, y = meters, fill = length)) +
  geom_col() +
  scale_fill_viridis_d()

p2 = ggplot(d2, aes(x = speed, y = value, fill = speed)) +
  geom_col()

p_ls = list(p1, p2)
n_x = sapply(p_ls, function(p) {
  max(layer_data(p, 1)$x)
})

plot_grid(plotlist = p_ls, align = 'v', ncol = 1, rel_widths = n_x)

enter image description here

Ответы [ 2 ]

2 голосов
/ 23 апреля 2020

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

Мой первый ответ (теперь второй вариант здесь) заключался в создании уровней ложных факторов. Это, безусловно, приносит идеальное выравнивание категорий.

Другим вариантом (теперь вариант 1 здесь) будет поиграться с аргументом расширения. Ниже програма c подходит к нему.

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

Но, в конце концов, я все же думаю, что вы можете получить более приятные и намного более легкие результаты с помощью огранки.

Один вариант

library(ggplot2)
library(cowplot)

d1 = data.frame(length = c('large', 'medium', 'small'), meters = c(100, 50, 30))

d2 = data.frame(speed = c('high', 'slow'), value =c(200, 45))

d3 = data.frame(key = c('high', 'slow', 'veryslow', 'superslow'), value = 1:4)

n_unq1 <- length(d1$length)
n_unq2 <- length(d2$speed)
n_unq3 <- length(d3$key)
n_x <- max(n_unq1, n_unq2, n_unq3)
#p1 = 
expand_n <- function(n_unq){
  if((n_x - n_unq)==0 ){
  waiver()
} else {
  expansion(add = c(0.6, (n_x-n_unq+0.56)))
}
}

p1 <- 
  ggplot(d1, aes(x = length, y = meters, fill = length)) +
  geom_col() +
  scale_fill_viridis_d() +
  scale_x_discrete(expand= expand_n(n_unq1)) +
  annotate(geom = 'rect', xmin = n_unq1+0.5, xmax = Inf, ymin = -Inf, ymax = Inf, fill = 'white')

p2 <- 
  ggplot(d2, aes(x = speed, y = value, fill = speed)) +
  geom_col() +
  scale_fill_viridis_d() +
  scale_x_discrete(expand= expand_n(n_unq2)) +
  annotate(geom = 'rect', xmin = n_unq2+0.5, xmax = Inf, ymin = -Inf, ymax = Inf, fill = 'white')

p3 <- 
  ggplot(d3, aes(x = key, y = value, fill = key)) +
  geom_col() +
  scale_fill_viridis_d() +
  scale_x_discrete(expand= expand_n(n_unq3)) +
  annotate(geom = 'rect', xmin = n_unq3+0.5, xmax = Inf, ymin = -Inf, ymax = Inf, fill = 'white')

p_ls = list(p1, p2,p3)

plot_grid(plotlist = p_ls, align = 'v', ncol = 1)

Создано в 2020-04-24 пакетом представ (v0.3.0)

Опция 2 , создайте n уровней поддельного фактора до максимального уровня графика и затем используйте drop = FALSE. Здесь программный c подход к нему

library(tidyverse)
library(cowplot)

n_unq1 <- length(d1$length)
n_unq2 <- length(d2$speed)
n_unq3 <- length(d3$key)
n_x <- max(n_unq1, n_unq2, n_unq3)

make_levels <- function(x, value) {
  x[[value]] <- as.character(x[[value]])
  l <- length(unique(x[[value]]))

  add_lev <- n_x - l

  if (add_lev == 0) {
    x[[value]] <- as.factor(x[[value]])
    x
  } else {
    dummy_lev <- map_chr(1:add_lev, function(i) paste(rep(" ", i), collapse = ""))
    x[[value]] <- factor(x[[value]], levels = c(unique(x[[value]]), dummy_lev))
    x
  }
}

list_df <- list(d1, d2, d3)
list_val <- c("length", "speed", "key")

fac_list <- purrr::pmap(.l = list(list_df, list_val), function(x, y) make_levels(x = x, value = y))

p1 <-
  ggplot(fac_list[[1]], aes(x = length, y = meters, fill = length)) +
  geom_col() +
  scale_fill_viridis_d() +
  scale_x_discrete(drop = FALSE) +
  annotate(geom = "rect", xmin = n_unq1 + 0.56, xmax = Inf, ymin = -Inf, ymax = Inf, fill = "white") +
  theme(axis.ticks.x = element_blank())
p2 <-
  ggplot(fac_list[[2]], aes(x = speed, y = value, fill = speed)) +
  geom_col() +
  scale_fill_viridis_d() +
  scale_x_discrete(drop = FALSE) +
  annotate(geom = "rect", xmin = n_unq2 + 0.56, xmax = Inf, ymin = -Inf, ymax = Inf, fill = "white") +
  theme(axis.ticks.x = element_blank())
p3 <-
  ggplot(fac_list[[3]], aes(x = key, y = value, fill = key)) +
  geom_col() +
  scale_fill_viridis_d() +
  scale_x_discrete(drop = FALSE) +
  annotate(geom = "rect", xmin = n_unq3 + 0.56, xmax = Inf, ymin = -Inf, ymax = Inf, fill = "white") +
  theme(axis.ticks.x = element_blank())

p_ls <- list(p1, p2, p3)

plot_grid(plotlist = p_ls, align = "v", ncol = 1)

Создан в 2020-04-24 с помощью пакета prex (v0. 3,0)

0 голосов
/ 23 апреля 2020

С ?plot_grid:

rel_widths
(необязательно) Числовой вектор относительной ширины столбцов. Например, в сетке из двух столбцов rel_widths = c (2, 1) сделает первый столбец в два раза шире второго.

Аргумент rel_widths ничего не делает в сетка из одного столбца.

Скорее всего, вам придется вручную вызвать cowplot::draw_plot с соответствующими размерами, чтобы расположить графики там, где вы хотите.

...