Ulepszanie wykresu

Aby wykres był bardziej czytelny, trzeba osiągnąć dwa cele:

Posortować słupki w kolejności rosnącej.
Dodać do wykresu etykiety odpowiadające nazwom cech.

W tym celu skorzystamy z indeksowania w NumPy. Metoda .argsort() sortuje tablicę i zwraca indeksy posortowanych elementów. Użyjemy tych indeksów, aby zrealizować oba cele.

To ćwiczenie jest częścią kursu

Marketing Analytics: Przewidywanie rezygnacji klientów w Pythonie

Zobacz kurs

Instrukcje do ćwiczenia

Oblicz posortowane indeksy tablicy importances, używając np.argsort() na importances.
Utwórz posortowane etykiety, odwołując się do kolumn zbioru X i indeksując je przez sorted_index.
Utwórz wykres, indeksując importances przez sorted_index i podając argument tick_label=labels.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# 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()

Edytuj i uruchom kod