ggparty 节点图中的分组百分比

Question

我正在与 ggparty 合作，并尝试在partykit plot（使用 geom_node_plot）的每个终端节点处自定义条形图。 我希望某些条形（及其标签）的长度显示总预测案例的比例——其中该百分比中的分母仅来自该节点的案例，而不是整个数据集。

（这在 ggplot2 中是一个很常见的问题，但这里的困难是特定于在 ggparty 内部工作的。）

require(ggplot2)
require(partykit)
require(ggparty)
    
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(count / sum(count))
        ),
        geom_bar()
      ))

ggparty的小插图指出每个节点 plot 是“独立的”，这表明数据是隔离的，但这似乎不适用于count ，因为如您所见，百分比是数据集中所有数据的一部分。

我知道一个稍微替代的方法是使用prop而不是count / sum(count) ，这允许“by”参数指定分组。 这似乎很理想——但是这里可用的分组变量是什么，它将隔离节点？ 派对 nodeid 似乎不在每个 geom_bar 调用的环境中。

任何自定义的分组计算，例如使用 lapply，也需要使用一个变量，指定终端节点，或者以其他方式区分图——所以这似乎是关键问题。

感谢您的任何指导。

Answer 1

据我所知，节点的条被添加为多面 plot。 要按面板获取百分比，您基本上可以使用与标准 ggplot 相同的方法，即正如您已经猜到的，您必须考虑可以从PANEL获得的面板的分组：

library(ggparty)

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

panel_prop <- function(count, panel) {
  count / tapply(count, panel, sum)[as.character(panel)]
}

ggparty(wptree) +
  geom_edge() +
  geom_edge_label() +
  geom_node_splitvar() +
  geom_node_plot(gglist = list(
    aes(
      y = play,
      x = after_stat(panel_prop(count, PANEL))
    ),
    geom_bar()
  ))