НачатьНачать бесплатно

Визуализация кластеров с помощью matplotlib

Мы уже обсудили, что визуализация необходима для оценки сформированных кластеров и выявления закономерностей в данных. Теперь сосредоточимся на визуализации набора данных о посещаемости Comic-Con с помощью модуля matplotlib.

Данные хранятся в DataFrame библиотеки pandas — comic_con. Столбцы x_scaled и y_scaled содержат стандартизированные координаты X и Y людей в определённый момент времени. В столбце cluster_labels находятся метки кластеров. Объект связи хранится в переменной distance_matrix.

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

Кластерный анализ на Python

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

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

  • Импортируйте класс pyplot из модуля matplotlib под псевдонимом plt.
  • Определите словарь colors для двух меток кластеров: 1 и 2.
  • Постройте диаграмму рассеяния, в которой каждый кластер окрашен в соответствии со словарём colors.

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

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

# Import the pyplot class
____

# Define a colors dictionary for clusters
colors = {____:'red', ____:'blue'}

# Plot a scatter plot
comic_con.plot.scatter(x=____, 
                	   y=____,
                	   c=comic_con['cluster_labels'].apply(____))
plt.show()
Редактировать и запускать код