使用scikit包在Python中绘制集群区域的边界

前端之家收集整理的这篇文章主要介绍了使用scikit包在Python中绘制集群区域的边界前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

这是我在3属性(x,y,value)中处理数据聚类的简单示例.每个样本代表其位置(x,y)及其所属变量.

我的代码发布在这里:

x = np.arange(100,200,1)
y = np.arange(100,1)
value = np.random.random(100*100)

xx,yy = np.meshgrid(x,y)
xx = xx.reshape(100*100)
yy = yy.reshape(100*100)
j = np.dstack((xx,yy,value))[0,:,:]

fig = plt.figure(figsize =(12,4))
ax1 = plt.subplot(121)
xi,yi = np.meshgrid(x,y)
va    = value.reshape(100,100)
pc = plt.pcolormesh(xi,yi,va,cmap = plt.cm.Spectral)
plt.colorbar(pc)

ax2 = plt.subplot(122)
y_pred = KMeans(n_clusters=12,random_state=random_state).fit_predict(j)
vb = y_pred.reshape(100,100)
plt.pcolormesh(xi,vb,cmap = plt.cm.Accent)

这个数字如下:

enter image description here

如何识别每个聚类区域的边界并勾画它们以加强可视化效果.

PS

这是我手动绘制的插图.我需要的是识别聚类边界并用线条描绘它们.

enter image description here

PPS

我发现了一个有趣的问题here试图在R中绘制聚类区域的边界

enter image description here

更新

我尝试了子程序后:

for i in range(n_cluster):
    plt.contour(vb ==i contours=1,colors=['b']) 

完成!

enter image description here

最佳答案
群集区域实际上只是群集中心的Voronoi图. Scipy有一些工具可以在给定一组点的情况下计算Voronoi单元. This page有一些关于如何做到这一点的例子.

猜你在找的Python相关文章