CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Importez la classe pyplot du module matplotlib sous le nom plt.
  • Définissez un dictionnaire colors pour deux étiquettes de regroupement, 1 et 2.
  • 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()
Modifier et exécuter le code