Update plot_grid.R

Add the function which reoder the labels when byrow=F
wilkelab · Mar 13, 2024 · 1b133f4 · 1b133f4
1 parent e1334a2
commit 1b133f4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/R/plot_grid.R b/R/plot_grid.R
@@ -177,8 +177,10 @@ plot_grid <- function(..., plotlist = NULL, align = c("none", "h", "v", "hv"),
   if (is.null(cols)) cols <- ceiling(num_plots/rows)
   if (is.null(rows)) rows <- ceiling(num_plots/cols)
 
-  # if the user wants to layout the plots by column, we use the calculated rows to reorder plots
-  if (!isTRUE(byrow)) plots <- plots[c(t(matrix(c(1:num_plots, rep(NA, (rows * cols) - num_plots)), nrow = rows, byrow = FALSE)))]
+  # if the user wants to layout the plots by column, we use the calculated rows to reorder plots and labels
+  if (!isTRUE(byrow)) 
+    plots <- plots[c(t(matrix(c(1:num_plots, rep(NA, (rows * cols) - num_plots)), nrow = rows, byrow = FALSE)))]
+    labels <- labels[c(t(matrix(c(1:num_plots, rep(NA, (rows * cols) - num_plots)), nrow = rows, byrow = FALSE)))]
 
   # Align the plots (if specified)
   grobs <- align_plots(plotlist = plots, align = align, axis = axis, greedy = greedy)