Улучшение графика

Чтобы сделать график более наглядным, нужно решить две задачи:

Отсортировать столбцы по возрастанию.
Добавить на график метки, соответствующие названиям признаков.

Для этого воспользуемся индексацией NumPy. Метод .argsort() сортирует массив и возвращает индексы элементов. Эти индексы помогут нам решить обе задачи.

Это упражнение является частью курса

Маркетинговая аналитика: прогнозирование оттока клиентов на Python

Посмотреть курс

Инструкции к упражнению

Вычислите отсортированные индексы массива importances, применив np.argsort() к importances.
Создайте отсортированные метки, обратившись к столбцам X и выполнив индексацию по sorted_index.
Постройте график, проиндексировав importances с помощью sorted_index и указав именованный аргумент tick_label=labels.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

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

Редактировать и запускать код