python 特征重要性条形图

Question

I have created my feature importance bar charts with the following code. My question is how to label "Feature #" as the column name rather than feature 0, feature 1, etc. the first picture is what i made, the second picture is how i need it to look.

for i,v in enumerate(importance):
print('Feature: %0d, Score: %.5f' % (i,v))
plt.bar([x for x in range(len(importance))], importance)
plt.show()

my created feature selection bar chart image

how I want it to look image

Answer 1

You can use set_xtickslabel method in order to specify the horizontal legend. In your case, something like

ax1 = plt.subplot()
plt.bar([x for x in range(len(importance))], importance)
ax1.set_xticklabels([f'Feature {X}' for X in range(len(importance))) 

Check here for more info