如何在rpart.plot或fancyRpartPlot中使用带下标希腊字母标签？

核心解决方案 #

通过rpart.plot的text参数自定义节点文本，利用R的plotmath语法渲染带下标的希腊字母，避免直接修改数据框列名导致的解析问题。

完整代码示例 #

library(rpart.plot)
library(rpart)

# 构建数据集（保留普通变量名即可）
set.seed(1234)
df <- data.frame(
  nu = c(rnorm(mean = 1, n = 100), rnorm(mean = 2, n = 50)),
  rho = c(rnorm(n = 50), rnorm(sd = 2, n = 100)),
  cl = c(rep("TRUE", 100), rep("FALSE", 50))
)

# 训练决策树
tree <- rpart(cl ~ ., data = df)

# 自定义文本渲染函数，实现带下标希腊字母
rpart.plot(tree, 
           type = 4, 
           extra = 0, 
           branch.lty = 3, 
           box.palette = "RdYlGn",
           text = function(x) {
             # 为每个变量匹配对应的plotmath表达式
             var_exprs <- list(
               nu = expression(ν[1]),
               rho = expression(ρ[2])
             )
             # 节点为分裂节点时返回对应表达式，叶节点返回分类结果
             if (!is.na(x$var)) {
               var_exprs[[x$var]]
             } else {
               x$frame$yval2[x$node]
             }
           })

原理说明 #

1. 直接将数据框列名设为expression对象时，rpart会将其转换为字符串存储（如"ν[1]"），rpart.plot默认直接打印字符串，不会解析plotmath语法，因此无法显示下标。
2. rpart.plot的text参数允许传入自定义函数，函数接收包含节点信息的列表（如x$var为当前节点的分裂变量名），返回的expression对象会被R自动用plotmath渲染，从而正确显示带下标的希腊字母。

针对fancyRpartPlot的替代方案 #

fancyRpartPlot本质依赖rpart.plot的底层逻辑，若要实现相同效果，可直接使用上述rpart.plot的自定义文本方案，或修改fancyRpartPlot的调用参数传递text函数：

library(rattle)
fancyRpartPlot(tree, 
               type = 4, 
               extra = 0, 
               branch.lty = 3, 
               box.palette = "RdYlGn",
               text = function(x) {
                 var_exprs <- list(nu = expression(ν[1]), rho = expression(ρ[2]))
                 if (!is.na(x$var)) var_exprs[[x$var]] else x$frame$yval2[x$node]
               })

内容的提问来源于stack exchange，提问作者user3236841