Začněte nyníZačněte zdarma

Vizualizace clusterů pomocí matplotlib

Jak jsme si říkali, vizualizace jsou nezbytné k posouzení vzniklých clusterů a odhalení trendů v datech. Teď se zaměříme na vizualizaci datasetu návštěvnosti Comic-Conu pomocí modulu matplotlib.

Data jsou uložena v pandas DataFrame comic_con. x_scaled a y_scaled jsou názvy sloupců se standardizovanými souřadnicemi X a Y polohy lidí v daném okamžiku. cluster_labels obsahuje štítky clusterů. Linkage objekt je uložen v proměnné distance_matrix.

Toto cvičení je součástí kurzu

Cluster Analysis in Python

Zobrazit kurz

Pokyny k cvičení

  • Importuj třídu pyplot z modulu matplotlib jako plt.
  • Definuj slovník colors pro dva štítky clusterů: 1 a 2.
  • Vykresli bodový graf, kde každý cluster má barvu definovanou slovníkem colors.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# 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()
Upravit a spustit kód