from numpy import vstack,array

from numpy.random import rand

from scipy.cluster.vq import kmeans,vq,whiten

​

data = vstack((rand(100,3) + array([.5,.5,.5]),rand(100,3)))

​

data = whiten(data)

​

#设置 K = 3 (2 clusters) 计算 K-Means

centroids,_ = kmeans(data,3)

​

#将每个值分配给一个聚类

clx,_ = vq(data,centroids)

​

print(clx)