如何在ggh4x嵌套分面点图中保留指定空列以维持视觉对称

问题背景 #

基于训练组（TR）和对照组（UT）的基因表达数据绘制嵌套分面点图，展示基因通路在不同条件下的富集情况。分面维度包含训练状态、调控方向（UP/DOWN）、细胞标记群（Arc/Egr1/Jun），点大小映射enrichmentRatio，颜色映射-log10(FDR)。但使用ggh4x::facet_nested()并设置scales="free"时，无数据的细胞标记群列会被自动移除，导致部分分面缺少Arc/Egr1/Jun中的类别，破坏视觉对称，需要让每个分面强制保留全部3个细胞标记群列。

解决方法 #

核心思路是补全所有条件组合的空数据行，让ggplot识别到所有需要显示的类别，即使对应没有富集数据。具体步骤如下：

1. 生成所有条件的完整笛卡尔积
   用tidyr::expand_grid()创建训练状态、调控方向、细胞标记群的所有可能组合，再和原始数据左连接，保留所有组合（无数据的行对应值为NA）：

library(tidyr)
library(dplyr)

# 替换为你的原始数据对象
your_raw_data <- read.csv("your_enrichment_data.csv")

# 生成全部条件组合
full_conditions <- expand_grid(
  training_status = c("TR", "UT"),
  regulation = c("UP", "DOWN"),
  cell_cluster = c("Arc", "Egr1", "Jun")
)

# 左连接补全空组合
full_data <- left_join(full_conditions, your_raw_data,
                       by = c("training_status", "regulation", "cell_cluster"))

2. 修改绘图代码，保留所有分面类别
   使用补全后的full_data绘图，设置geom_point(na.rm = TRUE)忽略空值（避免报错且不显示无效点），分面参数用scales="free_x"配合space="free_x"，既保留所有x轴类别，又能优化布局：

library(ggplot2)
library(ggh4x)

ggplot(full_data, aes(x = cell_cluster, y = pathway_name,
                      size = enrichmentRatio, color = -log10(FDR))) +
  geom_point(na.rm = TRUE) +
  facet_nested(training_status ~ regulation,
               scales = "free_x", space = "free_x") +
  # 可根据需求调整大小和颜色刻度
  scale_size_continuous(range = c(1, 5)) +
  scale_color_viridis_c(option = "plasma") +
  theme_bw() +
  theme(axis.text.x = element_text(angle = 45, hjust = 1))

效果说明 #

处理后，每个嵌套分面（如TR-UP、UT-DOWN等）都会显示Arc、Egr1、Jun三个x轴类别，即使某些类别没有富集数据，也会保留对应的空列，完美维持视觉对称。na.rm = TRUE确保空值不会干扰绘图，仅保留有数据的点。

内容的提问来源于stack exchange，提问作者isaac vingan