如何在ggparty终端节点图中实现节点内分组百分比展示？

核心问题分析 #

你遇到的问题是after_stat(count / sum(count))计算的是整个数据集的占比，因为sum(count)在全局统计环境中计算，没有限定在当前节点的样本范围内。虽然ggparty每个节点的子图数据是独立的，但ggplot的统计变换默认不会自动按节点分组计算。

解决方案 #

关键是在节点子图的aes中指定group = 1，让统计变换（比如stat_count）基于当前节点的所有样本计算比例。这样after_stat(prop)就会返回当前节点内各分类的占比。

修改后的完整代码 #

require(ggplot2)
require(partykit)
require(ggparty)
require(scales) # 用于百分比格式转换

data("WeatherPlay", package = "partykit")
wptree <- ctree(play ~ outlook, data=WeatherPlay, control = ctree_control(minsplit=1, minbucket=1, mincriterion = .01))

ggparty(wptree) +
  geom_edge() +
  geom_edge_label() +
  geom_node_splitvar() +
  geom_node_plot(gglist = list(
    aes(
      y = play,
      x = after_stat(prop),
      group = 1  # 关键：让比例计算基于当前节点的所有样本
    ),
    geom_bar(stat = "count"),
    # 可选：添加百分比标签
    geom_text(
      aes(label = percent(after_stat(prop))),
      stat = "count",
      hjust = -0.1,
      size = 3
    ),
    xlim(c(0, 1.2))  # 调整x轴范围，避免标签被截断
  ))

代码说明 #

1. group = 1的作用：告诉ggplot的统计变换，计算比例时要把当前节点的所有样本作为一个整体，而不是默认按y=play分组。这样after_stat(prop)就会输出每个类别在当前节点中的占比，分母是该节点的样本总数。
2. stat = "count"：显式指定条形图用计数统计，配合after_stat(prop)直接获取节点内比例。
3. 百分比标签：用percent()将比例转换成百分比格式，hjust = -0.1让标签显示在条形右侧，xlim调整轴范围避免标签被截断。

替代方案：手动计算节点内比例 #

如果需要更灵活的控制，可以先为每个节点的数据集计算比例，再传入geom_node_plot：

require(dplyr)

ggparty(wptree) +
  geom_edge() +
  geom_edge_label() +
  geom_node_splitvar() +
  geom_node_plot(
    gglist = list(
      aes(y = play, x = prop),
      geom_bar(stat = "identity"),
      geom_text(aes(label = percent(prop)), hjust = -0.1, size = 3),
      xlim(c(0, 1.2))
    ),
    # 提前计算每个节点内的比例
    data = function(node_data) {
      node_data |>
        count(play) |>
        mutate(prop = n / sum(n))
    }
  )

这个方法通过data参数传入自定义函数，对每个节点的数据集先统计计数并计算比例，再用于绘图，适合需要额外数据处理的场景。

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