R 从ctree中提取内部节点信息并拆分（partykit）

您好，我目前正在尝试使用 partykit 中的 ctree 提取存储在 R 中常量派对对象中的一些内部节点信息，但我发现导航对象有点困难，我可以在绘图上显示信息，但是我不知道如何提取信息 - 我认为它需要 nodeapply 或 partykit 中的其他功能？

library(partykit)
irisct <- ctree(Species ~ .,data = iris)
plot(irisct, inner_panel = node_barplot(irisct))

包含内部节点详细信息的绘图 https://i.stack.imgur.com/RWwZc.png

所有信息都可以通过绘图函数访问，但我想要类似于以下内容的文本输出：输出示例 https://i.stack.imgur.com/wm9dA.gif

主要技巧（如@G5W之前指出的）是采取[id]的子集party对象，然后提取数据（通过$data或使用data_party()函数），其中包含响应。我建议首先构建一个包含绝对频率的表，然后从中计算相对频率和边际频率。使用irisct对象普通表可以通过以下方式获得

tab <- sapply(1:length(irisct), function(id) {
  y <- data_party(irisct[id])
  y <- y[["(response)"]]
  table(y)
})
tab
##            [,1] [,2] [,3] [,4] [,5] [,6] [,7]
## setosa       50   50    0    0    0    0    0
## versicolor   50    0   50   49   45    4    1
## virginica    50    0   50    5    1    4   45

然后我们可以添加一点格式到一个漂亮的table object:

colnames(tab) <- 1:length(irisct)
tab <- as.table(tab)
names(dimnames(tab)) <- c("Species", "Node")

然后使用prop.table() and margin.table()来计算我们感兴趣的频率。as.data.frame()方法变换自table布局为“长”data.frame:

as.data.frame(prop.table(tab, 1))
##       Species Node        Freq
## 1      setosa    1 0.500000000
## 2  versicolor    1 0.251256281
## 3   virginica    1 0.322580645
## 4      setosa    2 0.500000000
## 5  versicolor    2 0.000000000
## 6   virginica    2 0.000000000
## 7      setosa    3 0.000000000
## 8  versicolor    3 0.251256281
## 9   virginica    3 0.322580645
## 10     setosa    4 0.000000000
## 11 versicolor    4 0.246231156
## 12  virginica    4 0.032258065
## 13     setosa    5 0.000000000
## 14 versicolor    5 0.226130653
## 15  virginica    5 0.006451613
## 16     setosa    6 0.000000000
## 17 versicolor    6 0.020100503
## 18  virginica    6 0.025806452
## 19     setosa    7 0.000000000
## 20 versicolor    7 0.005025126
## 21  virginica    7 0.290322581

as.data.frame(margin.table(tab, 2))
##   Node Freq
## 1    1  150
## 2    2   50
## 3    3  100
## 4    4   54
## 5    5   46
## 6    6    8
## 7    7   46

并且可以通过（仍未导出）获得分割信息.list.rules.party()功能。你只需要询问all节点 ID（默认情况下仅使用终端节点 ID）：

partykit:::.list.rules.party(irisct, i = nodeids(irisct))
##                                                               1 
##                                                              "" 
##                                                               2 
##                                           "Petal.Length <= 1.9" 
##                                                               3 
##                                            "Petal.Length > 1.9" 
##                                                               4 
##                       "Petal.Length > 1.9 & Petal.Width <= 1.7" 
##                                                               5 
## "Petal.Length > 1.9 & Petal.Width <= 1.7 & Petal.Length <= 4.8" 
##                                                               6 
##  "Petal.Length > 1.9 & Petal.Width <= 1.7 & Petal.Length > 4.8" 
##                                                               7 
##                        "Petal.Length > 1.9 & Petal.Width > 1.7"