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
Instructions de l’exercice
- Importez
fclusteretlinkagedepuisscipy.cluster.hierarchy. - Utilisez la méthode
warddans la fonctionlinkage(). - 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()