import numpy as np
from sklearn import datasets
from sklearn.cluster import KMeans

X, y = datasets.load_iris(return_X_y=True)

nSample, nDim = X.shape
model = KMeans(n_clusters=3)
y_hat = model.fit_predict(X)
labs = np.unique(y_hat)

means = np.zeros((len(labs),nDim))
for i, lab in enumerate(labs):
    means[i] = np.mean(X[y_hat == lab], axis=0)
print(means)



 km = KMeans(n_clusters=3).fit(X)
 print(km.cluster_centers_)