Améliorer le graphique
Pour rendre le graphique plus lisible, nous devons atteindre deux objectifs :
- Réorganiser les barres par ordre croissant.
- Ajouter des étiquettes au graphique correspondant aux noms des variables.
Pour cela, nous allons utiliser l’indexation NumPy. La méthode .argsort() trie un tableau et renvoie les indices. Nous utiliserons ces indices pour atteindre les deux objectifs.
Cet exercice fait partie du cours
Marketing Analytics : prédire l’attrition client en Python
Instructions
- 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 spécifiant l’argument nommétick_label=labels.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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()