如何用ggplot绘制聚类分组的个体子图并解决空图问题

你遇到的空图问题，根源在于facet_grid(ID~cluster)会生成所有ID和cluster的交叉组合，但你的每个ID只属于一个特定聚类，那些不存在的组合自然就显示为空图了。下面给你几种优雅的解决方法：

方法一：重编码聚类内的ID（基础ggplot实现） #

我们可以给每个聚类内的ID重新分配本地编号（比如每个聚类里的ID都变成1、2），这样facet_grid就只会生成存在的组合：

# 加载dplyr用于数据处理
library(dplyr)
library(ggplot2)

# 处理数据：为每个聚类内的ID分配本地编号
df_processed <- df %>%
  group_by(cluster) %>%
  # 把每个聚类里的ID转换成1、2的因子
  mutate(local_id = as.factor(ID)) %>%
  mutate(local_id = forcats::fct_recode(local_id, 
                                       !!!set_names(1:n_distinct(local_id), unique(local_id)))) %>%
  ungroup()

# 绘图
ggplot(df_processed, aes(date, val)) +
  geom_bar(stat = 'identity') +
  facet_grid(local_id ~ cluster, scales = "free_y") +
  labs(x = "Date", y = "Value") +
  theme_bw()

这个方法不需要额外安装包，通过重编码ID让分面的交叉组合都是有效存在的，完美消除空图，同时保持2行3列的布局。

方法二：使用ggh4x的嵌套分面（更简洁的专业工具） #

ggh4x包扩展了ggplot的分面功能，支持嵌套分面自动过滤空组合，非常适合你的场景：

# 安装并加载ggh4x
install.packages("ggh4x")
library(ggh4x)
library(ggplot2)

# 直接使用嵌套分面
ggplot(df, aes(date, val)) +
  geom_bar(stat = 'identity') +
  facet_nested(ID ~ cluster, scales = "free_y") +
  labs(x = "Date", y = "Value") +
  theme_bw()

facet_nested会识别ID和cluster的嵌套关系（每个ID只属于一个cluster），自动只绘制存在的组合，不需要修改原始数据，代码更简洁。

方法三：手动组合子图（灵活定制布局） #

如果你需要更灵活的布局控制，可以用gridExtra手动为每个聚类创建子图再组合：

# 安装并加载gridExtra
install.packages("gridExtra")
library(gridExtra)
library(ggplot2)

# 为每个聚类生成包含对应ID子图的绘图对象
cluster_plots <- split(df, df$cluster) %>%
  lapply(function(cluster_data) {
    ggplot(cluster_data, aes(date, val)) +
      geom_bar(stat = 'identity') +
      facet_wrap(~ID, ncol = 1) +
      ggtitle(paste("Cluster", unique(cluster_data$cluster))) +
      theme_bw()
  })

# 组合成3列布局
grid.arrange(grobs = cluster_plots, ncol = 3)

这个方法适合需要对每个聚类的子图做单独定制的场景，完全避免空图问题。

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