如何在 Python 中使用 K-Means 聚类划分数据

Question

我使用 K-Means 聚类算法将我的数据分成 4 个部分。我将其可视化，但现在我需要每个集群作为数据框。有什么办法吗？

Answer 1

如果您使用 SKlearn 进行聚类，则可以使用 predict 对每个 cluster 的每个数据点进行分类并将其保存到字典中（作为例如）。

然后你可以使用from_records方法从Pandas到convert字典到DataFrame。