使用matplotli包对kmeans聚类结果可视化
聚类使用sklearn包中的kmeans算法,数据使用numpy随机生成
数据生成代码如下:
1.随机生成100*2的numpy二维数组
da1=np.random.rand(100,2)
2.原数据散点图展示
x=da1[:,0]
y=da1[:,1]
plt.scatter(x,y,s=16)
散点图如下:
3.调用kmeans包聚类
#kmeans包的引入
from sklearn.cluster import KMeans
#调用
km = KMeans(n_clusters=3, random_state=0)
y_p2 = km.fit_predict(da1)
4.聚类结果散点图
plt.rcParams['font.sans-serif'] = ['SimHei']
# 使负号可以展示
plt.rcParams['axes.unicode_minus'] = False
plt.scatter(da1[y_p2==0,0