如何使用文本绘制因子变量 rpart 的水平而不是标签/索引？

Question

在因子变量上绘制树分割图会生成一个图，其中节点标记有因子索引而不是级别文本。如何绘制标签为因子变量实际水平的树？

例如，这里是关于足球运动员的示例数据。这个想法是国家（通过目标变量）解释了玩家的收入。国家被用作因子变量。

goals <- rpois(n = 5,lambda = 2)
income <- rnorm(100, goals*100000, sd = 1+goals*1000)
country <- rep(c("England","USA", "Nigeria", "Argentia", "Belgium"),
               times = 20)
player_df <- data.frame(income = income, country = country)

t1 <- rpart(income ~ ., 
      data = player_df, 
      cp = 0.01)

plot(t1); text(t1,)

这是结果：

Answer 1

试试text(t1, pretty=FALSE)。这将防止因子标签被缩短。