如何使用 Scikit-learn 查找簇质心

Question

我有一个带有（标记的）集群的数据集。我试图找到每个簇的质心（一个向量，他的距离是簇的所有数据点中最小的）。

我找到了很多执行聚类的解决方案，然后才找到质心，但我还没有找到现有的。

Python schikit-learn 是首选。谢谢。

Answer 1

直接来自 docs:

from sklearn.neighbors.nearest_centroid import NearestCentroid
import numpy as np
X = np.array([[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]])
y = np.array([1, 1, 1, 2, 2, 2])
clf = NearestCentroid()
clf.fit(X, y)

print(clf.centroids_)
# [[-2.         -1.33333333]
#  [ 2.          1.33333333]]