python简单实现KMeans

1、安装sklearn
pip install sklearn
需要用到的库sklearn.cluster中的KMeans

2、案例分析:


# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans

x=[[98,90],
   [87,95],
   [45,54],
   [67,77],
   [23,12],
   [34,45],
   [78,77],
   [90,89],
   [67,69],
   [34,56]]

#print x

clf=KMeans(n_clusters=3)#分为三类
#print clf
pre =clf.fit_predict(x)#模型训练及预测,注意:只有聚类可以这么写
plt.title(u"KMeans Datamining")
#注意:plt.xlabel(u"英语")中的U是识别中文,要不然报错
plt.xlabel(u"英语")
plt.ylabel(u"语文")
#print pre
m=[k[0] for k in x]#遍历英语
n= [k[1] for k in x]#遍历语文
#print n
#print m
plt.scatter(m,n,c=pre,s=100,marker='x')#c=pre三种颜色
plt.show()

运行如下:
在这里插入图片描述

感悟:
不羡慕别人的辉煌,不喟叹世态炎凉,用平常的心态,经营最美好的生活。加油!everyday!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐