码迷,mamicode.com
首页 > 编程语言 > 详细

聚类--K均值算法:自主实现与sklearn.cluster.KMeans调用

时间:2018-11-10 12:59:45      阅读:261      评论:0      收藏:0      [点我收藏+]

标签:类别   load   com   enter   数据   att   lis   mat   RKE   

1.用python实现K均值算法

      1)选取初始数据中的k个对象作为初始的中心,每个对象代表一个聚类中心:

#随机生成一组整数sample
import numpy as np
sample=np.random.randint(1,100,[50,1])  
k=3#要分成的类别数
y=np.zeros(50)
#定义一个函数来存放开始的聚类中心kc
def start_center(sample,k):
    return sample[:3]
kc=start_center(sample,k)

  

2) 对于样本中的数据对象,根据它们与这些聚类中心的欧氏距离,按距离最近的准则将它们分到距离它们最近的聚类中心(最相似)所对应的类;

  def nearest(kc, x[i]): j

  def xclassify(x, y, kc):y[i]=j

?def nearest(kc,i):
    d=abs(kc-i)
    t=np.where(d ==np.min(d))
    return t[0][0]
nearest(kc,34)
#对其进行对应的分类
?def xclassify(sample,y,kc):
    for i in range(30):
        y[i]=nearest(kc,sample[i])
    return y
y=xclassify(sample,y,kc)
print(kc,y)

  

3) 更新聚类中心:将每个类别中所有对象所对应的均值作为该类别的聚类中心,计算目标函数的值;

  def kcmean(x, y, kc, k):

def kcmean(sample,y,kc,k):
    l=list(kc)
    flag=False
    for j in range(k):#对k的类别进行遍历
        m=np.where(y==j)
        print(j,sample[j])
        junzhi=np.mean(sample[m])#求出每个类别的均值
        print(kc[j],junzhi)
        if l[j]!=junzhi: #判断求出的均值是否与开始的看类别值相等
            l[j]=junzhi
            flag=True
    return(np.array(l),flag)

  

4) 判断聚类中心和目标函数的值是否发生改变,若不变,则输出结果,若改变,则返回2)。

  while flag:

      y = xclassify(x, y, kc)

      kc, flag = kcmean(x, y, kc, k)

flag=True
while flag:
    y=xclassify(sample,y,kc)
    kc,flag=kcmean(sample,y,kc,k)
    print(y,kc)
print(sample,y)

  技术分享图片

2. 鸢尾花花瓣长度数据做聚类并用散点图显示。

import matplotlib.pyplot as plt
import numpy as np
data=load_iris() data_length=data[‘data‘][:,2]#取出鸢尾花花瓣的长度 x=data_length y=np.zeros(x.shape[0]) kc = start_center(x,3) flag=True while flag: y=xclassify(x,y,kc) kc,flag=kcmean(x,y,kc,k) print(y,kc) #用散点图来显示 plt.scatter(x,x,s=x,c=y,cmap=‘rainbow‘,alpha=0.5,linewidths=4) plt.show()

  技术分享图片

3.用sklearm包的kmeans对鸢尾花的花瓣进行分析用散点图显示。

from sklearn.datasets import load_iris
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt
import numpy as np
# 用sklearn.cluster.KMeans,鸢尾花花瓣长度数据做聚类并用散点图显示.
data=load_iris()
data_length=data[‘data‘][:,2:3]#取出鸢尾花花瓣的长度
x=data_length
#y=np.zeros(x.shape[0])
k1=KMeans(n_clusters=3)#将其类别分为3类
k1.fit(x)
kc1=k1.cluster_centers_
y_kmeans=k1.predict(x)##预测每个样本的聚类索引
print(y_kmeans,kc1)
plt.scatter(x,np.linspace(1,150,150),c=y_kmeans,marker=‘x‘,cmap=‘rainbow‘,linewidths=4)
plt.show()

  技术分享图片

4.完整的鸢尾花数据分析用散点图显示

技术分享图片

 

聚类--K均值算法:自主实现与sklearn.cluster.KMeans调用

标签:类别   load   com   enter   数据   att   lis   mat   RKE   

原文地址:https://www.cnblogs.com/SpaldingWen/p/9938603.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!