Förbättra diagrammet
För att göra diagrammet mer lättläst behöver vi uppnå två saker:
- Sortera staplarna i stigande ordning.
- Lägga till etiketter i diagrammet som motsvarar särdragsnamnen.
Vi använder NumPy-indexering för att klara det. Metoden .argsort() sorterar en array och returnerar indexen. Vi använder dessa index för att uppnå båda målen.
Den här övningen är en del av kursen
Marknadsföringsanalys: Prediktera kundbortfall i Python
Övningsinstruktioner
- Beräkna de sorterade indexen för
importancesgenom att användanp.argsort()påimportances. - Skapa de sorterade etiketterna genom att hämta kolumnerna i
Xoch indexera medsorted_index. - Skapa diagrammet genom att indexera
importancesmedsorted_indexoch ange nyckelordsargumentettick_label=labels.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Sort importances
sorted_index = ____(____)
# Create labels
labels = X.columns[____]
# Clear current plot
plt.clf()
# Create plot
plt.barh(range(X.shape[1]), importances[____], tick_label=____)
plt.show()