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()
  ))