我在形状为[1000，]的一维数组中有一些数据，其中包含1000个元素。我对该数据应用了k均值聚类，聚类数为10。应用k均值后，我为每个聚类得到了形状为[1000]的聚类标签（id），形状为[10]的质心。标签将0到9之间的值分配给1000个元素中的每个元素。但是，我希望每个元素都显示其质心，而不是其簇ID。我该如何实现？
from sklearn.cluster import KMeans
kmeans = KMeans(n_clusters=10)
kmeans.fit(data) #data is of shape [1000,]
#learn the labels and the means
labels = kmeans.predict(data) #labels of shape [1000,] with values 0<= i <= 9
centroids = kmeans.cluster_centers_ #means of shape [10,]
在上面的代码中，我想要[1000，]数组中每个元素的质心，而不是其簇ID。