Améliorer le graphique
Pour rendre le graphique plus lisible, nous devons atteindre deux objectifs :
- Réorganiser les barres en ordre croissant.
- Ajouter des étiquettes au graphique qui correspondent aux noms des variables.
Pour y parvenir, nous allons exploiter l'indexation NumPy. La méthode .argsort() trie un tableau et renvoie les indices. Nous utiliserons ces indices pour atteindre les deux objectifs.
Cette activité fait partie du cours
Analytique marketing : prédire l'attrition de la clientèle avec Python
Instructions de l’exercice
- Calculez les indices triés de
importancesen utilisantnp.argsort()surimportances. - Créez les étiquettes triées en accédant aux colonnes de
Xet en indexant avecsorted_index. - Créez le graphique en indexant
importancesavecsorted_indexet en précisant l'argument nommétick_label=labels.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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()