如何使用scikit learn/pandas/python打印任何一個集羣的樣本/觀察/行？

Question

非常基本的問題：

如何打印屬於任何一個羣集的觀測值？

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_ 

我只想看看'Var1'，'Var2'，'Var3'的值對於任何一個羣集。

Answer 1

這聽起來像你想查看你的數據集的預測。這是可以做到如下：

y_pred = ms.predict(X) 

然後，您可以做這件事，這些鏈接到你的樣品：

df['y_pred'] = y_pred 

，並根據需要篩選數據幀，例如看到集羣1中的樣本：

df[df.y_pred == 1]