rpart.plot中的颜色节点(按颜色匹配节点列表)

library(rpart)
library(rpart.plot)

# 创建一个示例数据集
data <- iris

# 构建决策树模型
model <- rpart(Species ~ ., data = data)

# 绘制决策树，并设置颜色节点
plot(model, uniform = TRUE)
rpart.plot(model, type = 2, extra = 2, box.palette = c("red", "green", "blue"),
           box.col = "gray", nn = TRUE, yesno = 2, fallen.leaves = TRUE,
           fallen.leaves.col = "orange", fallen.leaves.lty = 2, fallen.leaves.lwd = 2,
           fallen.leaves.pch = 4, fallen.leaves.cex = 1, fallen.leaves.cex.corner = 0.8,
           fallen.leaves.edge = 0.05, fallen.leaves.edge.col = "gray")