如何获取剪枝后RPART树中所选特征的名称？

嘿，这个问题我太熟了——完全不用费劲从绘制的树里挨个读特征名！给你两个简单靠谱的方法，直接拿到你要的特征名称向量：

方法1：从树的内部结构提取（最直观可靠） #

当你用prune()得到剪枝后的树对象（比如叫pruned_tree），可以直接访问它的frame属性——这是rpart树存储节点信息的核心数据框，其中var列记录了每个节点拆分时用到的特征，叶子节点会标记为<leaf>。我们只需要筛选掉叶子节点的标记，再去重就能得到所有用到的特征：

# 假设你的剪枝树对象是 pruned_tree
used_features <- unique(pruned_tree$frame$var[pruned_tree$frame$var != "<leaf>"])
# 输出结果就是特征名称向量
used_features

这个方法直接从树的底层数据提取，不会有遗漏，而且自动帮你去掉重复的特征名（毕竟同一个特征可能在多个节点被用来拆分）。

方法2：利用变量重要性快速提取（更简洁） #

rpart树默认会计算每个特征的重要性（如果构建树时没开这个功能，记得加importance=TRUE参数，比如rpart(y ~ ., data=your_data, importance=TRUE)）。剪枝后的树依然保留这个属性，且只有被用到的特征才会有非零的重要性值，所以直接取这个属性的名称就行：

used_features <- names(pruned_tree$variable.importance)

这个方法输出的向量会按特征的重要性从高到低排序，如果你需要按重要性筛选或者排序的话，这个方法会更顺手。

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