如何简化ggmosaic添加计数的代码并提升其复用性？

嘿，我完全懂你说的这个痛点——用ggmosaic画带计数的马赛克图时，把绘图步骤拆成临时变量的写法，不仅繁琐，还违背了ggplot2一贯的链式语法风格。下面给你几个能大幅提升代码复用性的改进思路，附上手写示例：

1. 封装成可复用的自定义函数 #

把马赛克图的绘制逻辑打包成函数，只需要传入数据集、分类变量、标签参数等，就能一键生成符合需求的图。这是提升复用性最直接的方式，还能统一所有图的风格。

示例代码：

library(tidyverse)
library(ggmosaic)

# 封装带计数的马赛克图函数
plot_mosaic_with_counts <- function(data, x_var, y_var, fill_var = y_var, 
                                    plot_title = "马赛克图", count_text_size = 3) {
  # 提前计算分组计数（也可以在stat里直接调用，提前算更灵活调整位置）
  count_df <- data %>% count({{x_var}}, {{y_var}})
  
  ggplot(data) +
    # 绘制马赛克图主体
    geom_mosaic(aes(x = product({{x_var}}), fill = {{fill_var}}, 
                    conds = product({{y_var}})), 
                divider = mosaic("v"), color = "white") +
    # 添加计数标签
    geom_text(data = count_df, 
              aes(x = as.numeric({{x_var}}) - 0.4,  # 微调标签位置让它居中
                  y = as.numeric({{y_var}}) - 0.4,
                  label = n),
              size = count_text_size) +
    # 自动生成轴标签和标题
    labs(title = plot_title, 
         x = rlang::as_name(enquo(x_var)), 
         y = rlang::as_name(enquo(y_var)), 
         fill = rlang::as_name(enquo(fill_var))) +
    theme_minimal()
}

# 测试函数（用mtcars数据集举例）
plot_mosaic_with_counts(mtcars, cyl, am, fill_var = am, plot_title = "气缸数 vs 变速箱类型")

2. 利用stat_mosaic的内置计算直接加标签 #

其实ggmosaic的stat_mosaic本身会自动计算每个单元格的位置和计数，我们可以直接用它来生成标签，完全不需要单独准备计数数据，也不用临时变量：

ggplot(mtcars) +
  # 绘制马赛克图矩形
  stat_mosaic(aes(x = product(cyl), fill = am, conds = product(am)),
              divider = mosaic("v"), geom = "rect", color = "white") +
  # 直接调用stat计算的计数n作为标签
  stat_mosaic(aes(x = product(cyl), conds = product(am),
                  label = after_stat(n)),
              geom = "text", size = 3) +
  labs(title = "气缸数 vs 变速箱类型", x = "气缸数", y = "变速箱类型", fill = "变速箱类型") +
  theme_minimal()

这种写法完全贴合ggplot2的图层叠加逻辑，没有多余的临时变量，代码简洁又易读。

3. 统一默认样式减少重复代码 #

如果你经常画这类图，可以提前设置ggmosaic的默认样式和主题，避免每次都重复写主题、颜色参数：

# 设置mosaic几何对象的默认样式
update_geom_defaults("mosaic", list(color = "white", size = 0.5))
# 统一全局主题
theme_set(theme_minimal() + theme(panel.grid = element_blank()))

# 之后调用函数或绘图时，直接用默认样式即可
plot_mosaic_with_counts(mtcars, gear, carb, plot_title = "档位 vs 化油器数量")

为什么这些方案更优？ #

• 告别临时变量：所有逻辑都在一个ggplot链式调用或函数里，符合ggplot2的设计理念
• 复用性拉满：函数可以快速适配不同数据集和变量，改个参数就能出图
• 可读性更强：代码结构清晰，绘图主体和标签逻辑分离但又高度整合

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