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

K-means聚类算法

时间:2019-09-28 14:41:31      阅读:120      评论:0      收藏:0      [点我收藏+]

标签:start   最小   apr   shape   并且   实现   panda   不同的   初始   

   K-means聚类算法

1、算法思想

首先,我们可以随机选取K个对象作为初始的聚类中心,然后计算每个对象与每一个种子聚类中心的距离,然后分别把这些对象分配给距离对象最近的一个聚类中心,只要对象被分配以后,被分配的对象就和聚类中心组成一个新的聚类,分配后就不能改变了,然后在根据每个聚类的聚类中心和对象的距离进行重新计算,并且一直重复这样的计算,直到没有对象可以被重新分配给不同的聚类或聚类中心不再发生变化或误差平方和局部已经最小,这样循环计算就会停止。

2、优缺点

优点:a.算法容易实现

           b.当聚类个数小于真实值时,误差平方和会下降的很快;

缺点:a.当聚类个数超过真实值时,误差平方和虽然会继续下降,但是下降的速度会缓减,而这个转折点就是最佳的聚类个数了。

           b.非均质的数据、不同类别内部方差不相同 

3、伪代码

输入:类簇的个数K,迭代终止值Z。

输出:聚类结果

For(t=1;t<K;t++){

     给出数据对象Xi;

     计算聚类中心和对象的距离dist(Xi,Centerk);

     将Xi划到距离Xi最近的类簇中心所在的类簇中

For(直到Xi不能在分配){

将所有的类簇中心更新

}

输出聚类结果;

}

4.代码实现

 

import numpy as np

 

import pandas as pd

 

import random

 

import sys

 

import time

 

class KMeansClusterer:

 

    def __init__(self,ndarray,cluster_num):

 

        self.ndarray = ndarray

 

        self.cluster_num = cluster_num

 

        self.points=self.__pick_start_point(ndarray,cluster_num)

 

         

 

    def cluster(self):

 

        result = []

 

        for i in range(self.cluster_num):

 

            result.append([])

 

        for item in self.ndarray:

 

            distance_min = sys.maxsize

 

            index=-1

 

            for i in range(len(self.points)):                

 

                distance = self.__distance(item,self.points[i])

 

                if distance < distance_min:

 

                    distance_min = distance

 

                    index = i

 

            result[index] = result[index] + [item.tolist()]

 

        new_center=[]

 

        for item in result:

 

            new_center.append(self.__center(item).tolist())

 

        # 中心点未改变,说明达到稳态,结束递归

 

        if (self.points==new_center).all():

 

            return result

 

         

 

        self.points=np.array(new_center)

 

        return self.cluster()

 

             

 

    def __center(self,list):

 

        ‘‘‘计算一组坐标的中心点

 

        ‘‘‘

 

        # 计算每一列的平均值

 

        return np.array(list).mean(axis=0)

 

    def __distance(self,p1,p2):

 

        ‘‘‘计算两点间距

 

        ‘‘‘

 

        tmp=0

 

        for i in range(len(p1)):

 

            tmp += pow(p1[i]-p2[i],2)

 

        return pow(tmp,0.5)

 

    def __pick_start_point(self,ndarray,cluster_num):

 

        

 

        if cluster_num <0 or cluster_num > ndarray.shape[0]:

 

            raise Exception("簇数设置有误")

 

      

 

        # 随机点的下标

 

        indexes=random.sample(np.arange(0,ndarray.shape[0],step=1).tolist(),cluster_num)

 

        points=[]

 

        for index in indexes:

 

            points.append(ndarray[index].tolist())

 

        return np.array(points)

 

 

 

 

代码参考资料cf44fxnaALrxBy0alj1s9pDTHymWapRwW4pnYyFhdnzSIAK-aAFBsCADRaJbBaRXne6CwU5lM4ucx-Y2YTUKejAT7qcyU_OsvFfLJldKJPmfADilL3a6K3QzB3u--M2y.htm

 

 

K-means聚类算法

标签:start   最小   apr   shape   并且   实现   panda   不同的   初始   

原文地址:https://www.cnblogs.com/wyf-1999-1--6/p/11602519.html

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