如何使用树林来标记特征重要性？

Question

我使用sklearn来绘制树木森林的特征重要性。数据帧名为“heart”。下面是提取已排序功能列表的代码：

importances = extc.feature_importances_
indices = np.argsort(importances)[::-1]
print("Feature ranking:")

for f in range(heart_train.shape[1]):
    print("%d. feature %d (%f)" % (f + 1, indices[f], importances[indices[f]]))

然后我以这种方式绘制列表：

f, ax = plt.subplots(figsize=(11, 9))
plt.title("Feature ranking", fontsize = 20)
plt.bar(range(heart_train.shape[1]), importances[indices],
    color="b", 
    align="center")
plt.xticks(range(heart_train.shape[1]), indices)
plt.xlim([-1, heart_train.shape[1]])
plt.ylabel("importance", fontsize = 18)
plt.xlabel("index of the feature", fontsize = 18)

我得到了这样一个情节：

我的问题是：如何用特征的名称替换特征的编号，以使绘图更容易理解？ 我试图转换包含功能名称的字符串（这是数据帧的每一列的名称），但无法达到我的目标。

谢谢

Answer 1

我知道这已经很古老了，但是对于后代来说，如果你想从@bakkal的解中以正确的顺序得到feature_name，你可以使用

feature_names = [features_names[i] for i in indices]