python - 如何将分离的部分(来自 k-means 聚类)保存为不同的数据帧

Question

我对包含 salesNumber 和 Products 的数据框使用了 k 均值聚类。我使用了 3 均值聚类。我可以将它们分开并找到质心。但我也想将新的 3 个不同的划分数据保存为 3 个不同的数据框。我怎样才能做到这一点？

df 数据框有 2 列，分别名为 SalesNumbers 和 ProductTypes。

kmeans = KMeans(n_clusters=3).fit(df)
centroidsProduct = kmeans.cluster_centers_
print(centroidsProduct)

Answer 1

使用labels_:

df['ClusterID'] = kmeans.labels_