kmeans聚类代码详解

kmeans聚类

keans算法主要是将所有点与质心算距离，与每个点距离最小的质心就是它的类别，然后再从每一类中选取平均值作为新的质心

#核心算法中的数据处理示例：
#假设五个点与四个之心距离分别如下：
dist = np.array([[112,324,532,32],
               [56,574,592,236],
               [13,34,57,32],
               [420,329,532,32],
               [112,324,52,32]])
#截取每一行的最小值
c_ind=np.argmin(dist,axis=1)
print(c_ind)
#打印类别是否是3
print(c_ind==3)
#打印类别为3的点
print(dist[c_ind==3])
#打印类别为3的新的质心
prin(np.mean(dist[c_ind==3],axis=0))

[3 0 0 3 3]
[ True False False True True]
[[112 324 532 32]
[420 329 532 32]
[112 324 52 32]]
array([214.66666667, 325.66666667, 372. , 32. ])

以下是详细代码：

import numpy as np
import matplotlib