ПочатиПочніть безкоштовно

Візуалізація кластерів з matplotlib

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

Дані збережено в датафреймі 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()
Редагувати та запускати код