ÎncepețiÎncepe gratuit

Vizualizează clusterele cu matplotlib

Am discutat că vizualizările sunt esențiale pentru a evalua clusterele formate și a identifica tendințe în date. Acum ne vom concentra pe vizualizarea setului de date privind fluxul de persoane de la Comic-Con, folosind modulul matplotlib.

Datele sunt stocate într-un DataFrame pandas, comic_con. x_scaled și y_scaled sunt numele coloanelor pentru coordonatele X și Y standardizate ale persoanelor la un moment dat. cluster_labels conține etichetele clusterelor. Un obiect de tip linkage este stocat în variabila distance_matrix.

Acest exercițiu face parte din cursul

Analiza clusterelor în Python

Vezi cursul

Instrucțiuni pentru exercițiu

  • Importă clasa pyplot din modulul matplotlib cu aliasul plt.
  • Definește un dicționar colors pentru cele două etichete de cluster, 1 și 2.
  • Trasează un grafic scatter cu culori pentru fiecare cluster, conform dicționarului colors.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# 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()
Editează și rulează codul