Visualiser des regroupements avec matplotlib
Nous avons vu que les visualisations sont essentielles pour évaluer les regroupements formés et repérer les tendances dans vos données. Concentrons-nous maintenant sur la visualisation de l'ensemble de données sur l'achalandage à Comic-Con à l'aide du module matplotlib.
Les données sont stockées dans un DataFrame pandas, comic_con. x_scaled et y_scaled sont les noms des colonnes correspondant aux coordonnées X et Y normalisées des personnes à un moment donné. cluster_labels contient les étiquettes de regroupement. Un objet de chaînage (linkage) est stocké dans la variable distance_matrix.
Cette activité fait partie du cours
Analyse de grappes en Python
Instructions de l’exercice
- Importez la classe
pyplotdu modulematplotlibsous le nomplt. - Définissez un dictionnaire
colorspour deux étiquettes de regroupement,1et2. - Tracez un nuage de points en appliquant une couleur à chaque regroupement selon le dictionnaire
colors.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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()