CommencezCommencez gratuitement

Regroupement hiérarchique : méthode de Ward

C'est le temps du Comic-Con! Le Comic-Con est un congrès annuel consacré aux bandes dessinées, tenu dans de grandes villes à travers le monde. Vous disposez des données de l'achalandage de l'an dernier, soit le nombre de personnes présentes sur le site à un moment donné. Vous voulez choisir l'emplacement de votre kiosque pour maximiser les ventes. En utilisant la méthode de Ward, appliquez un regroupement hiérarchique pour repérer deux points d'attraction dans la zone.

Les données sont stockées dans un DataFrame pandas, comic_con. x_scaled et y_scaled sont les noms des colonnes des coordonnées X et Y normalisées des personnes à un moment donné.

Cette activité fait partie du cours

Analyse de grappes en Python

Voir le cours

Instructions de l’exercice

  • Importez fcluster et linkage depuis scipy.cluster.hierarchy.
  • Utilisez la méthode ward dans la fonction linkage().
  • Assignez les étiquettes de groupe en formant 2 groupes plats à partir de distance_matrix.
  • Exécutez le code de visualisation pour voir les résultats.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Import the fcluster and linkage functions
from scipy.cluster.hierarchy import ____, ____

# Use the linkage() function
distance_matrix = ____(comic_con[['x_scaled', 'y_scaled']], ____ = ____, metric = 'euclidean')

# Assign cluster labels
comic_con['cluster_labels'] = ____(____, ____, criterion='maxclust')

# Plot clusters
sns.scatterplot(x='x_scaled', y='y_scaled', 
                hue='cluster_labels', data = comic_con)
plt.show()
Modifier et exécuter le code