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
Instrucțiuni pentru exercițiu
- Importă clasa
pyplotdin modululmatplotlibcu aliasulplt. - Definește un dicționar
colorspentru cele două etichete de cluster,1și2. - 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()