K-means聚类算法实现分类_k-means聚类将自己的身份证号码聚类成3类

K-means属于无监督学习

from numpy import vstack
from scipy.cluster.vq import kmeans, vq

list1 = [88.0, 64.0, 96.0, 85.0]
list2 = [92.0, 99.0, 95.0, 94.0]
list3 = [91.0, 87.0, 99.0, 95.0]
list4 = [78.0, 99.0, 97.0, 81.0]
list5 = [88.0, 78.0, 98.0, 84.0]
list6 = [100.0, 95.0, 100, 92.0]

data = vstack((list1, list2, list3, list4, list5, list6))  # vstack: 堆积成绩数据
centroids, _ = kmeans(data, 3)  # 返回聚类中心， 2：表示分类的数目
result, _ = vq(data, centroids)  # vq函数用来获取每一位同学所属的类别
print(result)
1
2
3
4
5
6
7
8
9
10
11
12
13
14

输出结果：[2 0 0 1 2 0] 将6组数据分类