import pandas as pd
import numpy as np
from sklearn.cluster import KMeans
from collections import Counter
##Préparer le dataframe df###
num_clus = 4 #Définir le nombre de clusters
kmeans = KMeans(n_clusters=num_clus, random_state=0).fit(df)
print(Counter(kmeans.labels_)) #Sortie du nombre de personnes dans chaque cluster
df['cluster_id']=kmeans.labels_ #Ajouter le numéro de cluster à la trame de données d'origine
for i in range(0,num_clus): #Sortie de la valeur moyenne de chaque cluster
print(df[df['cluster_id']==i].mean())
Recommended Posts