在R中的ggplot中绘制树

Question

不知道ggplot可以画树吗？比方说：

library(rpart
library(rpart.plot)

data(iris)
mod <- rpart(Species~., data=iris)
prp(mod)

我可以在 ggplot 中绘制类似的图形吗？

Answer 1

survMisc 包中的 autoplot.rpart() 函数可以帮助您完成部分工作。但是您可能需要清理情节的呈现，可能在符号中分层等。这似乎只是一个起点：

library(rpart)
library(survMisc)

data(iris)
mod <- rpart(Species~., data=iris)

autoplot(mod, branch=0)

Answer 2

您可以使用 rattle，它会生成漂亮的 rpart 图

library(rattle)

fancyRpartPlot(rpart(Species~., data=iris),yesno=2,split.col="black",nn.col="black", 
               caption="",palette="Set2",branch.col="black")

在R中的ggplot中绘制树

plot tree in ggplot in R

tree

r

ggplot2