我有一组执行 KMeans 分类的点。如何根据它们所属的集群绘制点颜色的图？
编辑：为澄清起见，拥有一组点，我想使用从KMeans.predict()（来自 sklearn）生成的数组的值来选择每个点的颜色。
matplotlib 上基于簇的着色点
数据挖掘
Python
可视化
2022-03-01 15:25:04
1个回答
sklearn文档向您展示了如何：
colors = np.array([x for x in 'bgrcmykbgrcmykbgrcmykbgrcmyk'])
colors = np.hstack([colors] * 20)
...
if hasattr(algorithm, 'cluster_centers_'):
centers = algorithm.cluster_centers_
center_colors = colors[:len(centers)]
plt.scatter(centers[:, 0], centers[:, 1], s=100, c=center_colors)
其它你可能感兴趣的问题