from sklearn.cluster import KMeans
import numpy as np
import matplotlib.pyplot as plt
data = np.genfromtxt("kmeans.txt", delimiter=" ")
k = 4
model = KMeans(n_clusters=k)
model.fit(data)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/28b080310a3e62af1ccef702092bfde1.png)
centers = model.cluster_centers_
print(centers)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/92de3b57ad0cb960bf413345aa09ed17.png)
result = model.predict(data)
print(result)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/bca5dae5dfcd9cc67d9aba278236c96f.png)
model.labels_
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/8c1fc26780a16ee05d9c996b6355ed52.png)
mark = ['or', 'ob', 'og', 'oy']
for i,d in enumerate(data):
plt.plot(d[0], d[1], mark[result[i]])
mark = ['*r', '*b', '*g', '*y']
for i,center in enumerate(centers):
plt.plot(center[0],center[1], mark[i], markersize=20)
plt.show()
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/891cdf5984afc6435715efeaecda8799.png)
x_min, x_max = data[:, 0].min() - 1, data[:, 0].max() + 1
y_min, y_max = data[:, 1].min() - 1, data[:, 1].max() + 1
xx, yy = np.meshgrid(np.arange(x_min, x_max, 0.02),
np.arange(y_min, y_max, 0.02))
z = model.predict(np.c_[xx.ravel(), yy.ravel()])
z = z.reshape(xx.shape)
cs = plt.contourf(xx, yy, z)
mark = ['or', 'ob', 'og', 'oy']
for i,d in enumerate(data):
plt.plot(d[0], d[1], mark[result[i]])
mark = ['*r', '*b', '*g', '*y']
for i,center in enumerate(centers):
plt.plot(center[0],center[1], mark[i], markersize=20)
plt.show()
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/654d23e570c69b6e5c879e6440d12544.png)