使用 kmeans 计算簇质心答案

【问题标题】：calculate cluster centroid using kmeans使用 kmeans 计算簇质心
【发布时间】：2021-06-13 15:28:35
【问题描述】：

我想用 scikit-learn 计算一个簇的质心向量：

from sklearn.cluster import KMeans
import numpy as np

kmeans = KMeans(n_clusters=1, random_state=0).fit(X)

如果我设置n_cluster=1，这是否意味着我计算了X的平均值？

比如X中有3个数据点，每个数据点都有一个4d向量，X=[[1,2,3,4],[2,3,4,5],[6,7,8,9]]，那么我用kmeans计算出质心向量后就是centroid_vect = [3,4,5,6]？

【问题讨论】：

【解决方案1】：

是的，没错。在拟合模型后，质心存储在kmeans.cluster_centers_中。

kmeans = KMeans(n_clusters=1, random_state=0).fit([[1,2,3,4],[2,3,4,5],[6,7,8,9]])

kmeans.cluster_centers_
#array([[3., 4., 5., 6.]])

【讨论】：