Comment imprimer des échantillons/observations/lignes d'un cluster en utilisant scikit learn/pandas/python?

Question

Question très basique:

Comment imprimer les observations attribuées à un cluster?

data = pd.read_csv('my_file.csv') 
X = data[['Var1','Var2','Var3']] 
ms = MeanShift() 
ms.fit(X) 
labels = ms.labels_ 
cluster_centers = ms.cluster_centers_ 

Je voudrais voir les valeurs de « Var1 », « Var2 », « Var3 » pour une grappe.

Answer 1

Il semble que vous voulez voir les prédictions pour votre ensemble de données. Cela peut se faire comme suit:

y_pred = ms.predict(X) 

Vous pouvez ensuite les relier à vos échantillons en faisant ceci:

df['y_pred'] = y_pred 

Et filtrer votre dataframe comme vous le souhaitez, par exemple pour voir les exemples dans le groupe 1:

df[df.y_pred == 1]