Python:二维数据聚类实现。

# coding=utf-8

"""
https://blog.csdn.net/gusui7202/article/details/83239142
qhy。
"""

from pylab import *
from scipy.cluster.vq import *
 
# 添加中文字体支持
from matplotlib.font_manager import FontProperties
font = FontProperties(fname=r"c:\windows\fonts\SimSun.ttc", size=14)
#随机生成数据
class1 = 1.5 * randn(10000, 2)
class2 = randn(10000, 2)
#提取数据特征点
features = vstack((class1, class2))
print(features)#打印数据堆叠结果
centroids, variance = kmeans(features, 3)#得到聚类结果

print(centroids)打印聚类中心
code, distance = vq(features, centroids)
figure()
ndx = where(code == 0)[0]
plot(features[ndx, 0], features[ndx, 1], '*')
ndx = where(code == 1)[0]
plot(features[ndx, 0], features[ndx, 1], 'r.')
plot(centroids[:, 0], centroids[:, 1], 'go')
title(u'2维数据点聚类', fontproperties=font)
axis('off')
show()