sklearn之聚类算法

from sklearn.datasets import make_blobs
import matplotlib.pyplot as plt

x, y = make_blobs(n_samples=500, n_features=2, centers=4, random_state=42)

fig, ax1 = plt.subplots(1)
ax1.scatter(x[:,0],x[:,1],
           marker='o',
           s=8)
plt.show()

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-WXQaHVsR-1592795660603)(output_2_0.png)]

color = ['red','pink','orange','grey']
fig, ax1 = plt.subplots(1)

for i in range(4):
    ax1.scatter(x[y==i, 0], x[y==i, 1],
               marker='o',
               s=8,
               c=color[i])
plt.show()

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-tNR7DTp2-1592795660606)(output_3_0.png)]

from sklearn.cluster import KMeans
n_clusters = 3
cluster = KMeans(n_clusters = n_clusters, random_state=0).fit(x)

y_pred