python简单实现KMeans
·
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!
更多推荐



所有评论(0)