python - 如何知道每个簇标签对应于哪个矩阵行？

Question

完成聚类后，我最终得到一个存储所有聚类标签的对象，如下所示:

clusterer.labels_

上面通常是一个列表或数组。然后我总是将标签分配给原始的 pandas 数据框(数据集)，如下所示:

df['cluster_lables] = cluster.labels_

最后，我假设 cluster.labels_ 的每个元素对应于原始数据集的每一行，这个假设正确吗？例如，从上面的专栏创建中，我最终得到如下结果:

ColA ColB cluster_labels
1    3       -1
2    4         2
...
89  90        45

Answer 1

大体上是的，你是对的。我之前使用过的聚类类型是 KMeans 聚类(可以在此处找到 https://scikit-learn.org/stable/modules/generated/sklearn.cluster.KMeans.html )，但我不能保证它们都像那样工作。将新列附加到数据框将按照您的想法工作。