r - r 中决策树中的叠加直方图

Question

我想创建一个决策 TreeMap ，它在每个节点中显示叠加直方图，如图所示:

以下代码创建左侧图片:

library(earth)
library(rpart)
library(rpart.plot)

a <- rpart(O3~., data=ozone1, cp=.2)
prp(a, type=4, fallen=T, branch=.3, round=0, leaf.round=9,
    clip.right.labs=F, under.cex=1,
    box.palette="GnYlRd",
    prefix="ozone\n", branch.col="gray", branch.lwd=2,
    extra=101, under=T, lt=" < ", ge=" >= ", cex.main=1.5)

我认为 rpart.plot 中的 node.fun 可以完成这项工作，但我不知道如何实现它。

Answer 1

不幸的是，当前版本的 rpart.plot 2.1.2 无法做到这一点。

正如您所提到的，您也许可以使用自定义的node.fun来绘制直方图。但 rpart.plot 不知道如何计算直方图的绘制大小，因此它不知道如何在页面上正确定位绘图的元素。

我会将其添加到包 TODO 列表中并考虑实现它，但不幸的是这不会很快发生。