更改 rpart 图中分支名称的格式答案

【问题标题】：Change formatting for branch names in rpart plot更改 rpart 图中分支名称的格式
【发布时间】：2021-11-17 03:54:24
【问题描述】：

当使用 rpart 创建和绘制树时，有许多函数可以改变最终外观，但是它似乎没有内置任何允许格式化分支名称的功能。以下是 (A) 正常情况的示例，以及 (B) 尝试使用 split.fun 更改名称时的示例，以及生成此图的代码。

test <- list()
test$tree <- rpart(Species ~ ., data = iris)
par(mfrow = c(1,2))
rpart.plot(test$tree, type=5, extra=2)
rpart.plot(test$tree, type=5, extra=2, split.fun = function(x, labs, digits, varlen, faclen){
  labs <- gsub(".", " ", labs)
  labs
})

我所追求的是将Petal.Length 和Petal.Width 显示为Petal Length 和Petal Width。有没有什么代码可以完成这个看似简单的任务？

【问题讨论】：

您的 gsub 正则表达式将所有字符更改为空白。而是使用 gsub("\\.", " ", labs) 或 gsub(".", " ", labs, fixed=TRUE)，而不是 gsub(".", " ", labs)。

标签： r rpart

【解决方案1】：

为了得到你想要的东西，我提供了一个 hack。不漂亮，但它的工作。

如果您查看树结构，这些标签来自 test$tree$frame$var。所以你可以简单地改变树中的那些。

par(mfrow = c(1,2))
rpart.plot(test$tree, type=5, extra=2)
test$tree$frame$var = sub("\\.", " ", test$tree$frame$var)
rpart.plot(test$tree, type=5, extra=2)

【讨论】：