Scikit learn 使用kmeans计算簇质心

我想用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中有三个数据点，每个数据点都有一个4d向量，

X=[[1,2,3,4]，[2,3,4,5]，[6,7,8,9]

，那么在我使用kmeans计算质心向量之后，它将是

质心向量=[3,4,5,6]

？

是的，准确地说。质心存储在

kmeans.cluster\u centers\uz

拟合模型后

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.]])