# -*- coding: utf-8 -*-
from sklearn.cluster import KMeans
import numpy as np
import matplotlib.pyplot as plt
#初始化数据集————二维
data=np.random.rand(1000,2)
print(data)
#指定簇的个数，即分成几类
km=KMeans(n_clusters=3).fit(data)
#标签结果
rs_labels=km.labels_
# print(rs_labels)
#每个类别的中心点
rs_center_ids=km.cluster_centers_
# # 第1维数据
# print(data[:,0])
# # 第2维数据
# print(data[:,1])
#描绘各个点
plt.scatter(data[:,0],data[:,1],c=rs_labels,alpha=0.5)
#描绘质心
plt.scatter(rs_center_ids[:,0],rs_center_ids[:,1],c='red')
plt.show()
-
运行结果：
运行结果
网友评论