r - 如何从ctree函数中提取树结构？

Question

我正在尝试从 ctree 的输出中提取树信息。我尝试了“BinaryTree”类信息，但没有成功。任何输入表示赞赏。

谢谢你

Answer 1

ctree 对象至少在顶部是 S4 对象，并且树信息在“树”槽中。 “树槽可以用@ 操作符访问。如果你在帮助(ctree)页面中使用第一个例子，你可以获得图形显示:

plot(airct)

然后你可以通过遍历列表操作来查看树的分支。树的“叶子”是“终端”==TRUE 节点的后代:

> airct@tree$right$terminal
[1] FALSE
> airct@tree$left$terminal
[1] FALSE
> airct@tree$right$right$terminal
[1] TRUE
> airct@tree$right$left$terminal
[1] TRUE
> airct@tree$left$left$terminal
[1] TRUE
> airct@tree$left$right$terminal
[1] FALSE

叶子上方节点的信息也可以恢复:

> airct@tree$left$right
4) Temp <= 77; criterion = 0.997, statistic = 11.599
  5)*  weights = 48 
4) Temp > 77
  6)*  weights = 21 

这与 nodes 的信息相同如果您知道节点的数量，函数将恢复:

> nodes(airct,4)
[[1]]
4) Temp <= 77; criterion = 0.997, statistic = 11.599
  5)*  weights = 48 
4) Temp > 77
  6)*  weights = 21