聚类分析算法Python3.6实践K均值聚类（K-means）

最新推荐文章于 2024-08-22 14:09:21 发布

liouwuzhou_bk

最新推荐文章于 2024-08-22 14:09:21 发布

阅读量4.5k

点赞数

文章标签：聚类分析算法 Python3.6 K-means

本文链接：https://blog.csdn.net/m0_37647623/article/details/78350289

版权

本文介绍了如何将Python2.7.5版本的聚类分析算法K-means移植到Python3.6，并提供了修改后的代码，同时为了便于理解，还展示了原始数据的部分绘图。

摘要由CSDN通过智能技术生成

在http://blog.csdn.net/zouxy09/article/details/17589329 上看到聚类分析算法，但是是基于python2.7.5版本，直接移植到Python3.6会有问题，更改代码后为增加对比性，绘制原始数据不部分；代码如下：

#################################################
#pyhon3.6运行
#time:20170501
#version
#liouwuzhou
#Email  : liouwuzhou@139.com
#################################################


def euclDistance(vector1, vector2):
    return sqrt(sum(power(vector2 - vector1, 2)))
def initCentroids(dataSet, k):
    numSamples, dim = dataSet.shape
    centroids = zeros((k, dim))
    for i in range(k):
        index = int(random.uniform(0, numSamples))
        centroids[i, :] = dataSet[index, :]
    return centroids

def kmeans(dataSet, k):
    numSamples = dataSet.shape[0]
    clusterAssment = mat(zeros((numSamples, </