我对包含 salesNumber 和 Products 的数据帧使用了 k 均值聚类分析。我使用了 3 表示聚类。我可以将它们分开并找到质心。但我也想将我的新 3 个不同的分割数据保存为 3 个不同的数据帧。我该怎么做?
df 数据帧有 2 列,分别名为"销售数字"和"产品类型"。
kmeans = KMeans(n_clusters=3).fit(df)
centroidsProduct = kmeans.cluster_centers_
print(centroidsProduct)
使用labels_
:
df['ClusterID'] = kmeans.labels_