【发布时间】:2016-06-01 08:43:50
【问题描述】:
我在 numpy 数组中有以下内容。
[[5 1 1 2]
[2 1 1 2]
[5 2 1 1]
[4 1 0 2]
[6 1 0 1]
[6 6 3 2]]
我使用以下代码执行 k-means 聚类
kmeans=KMeans(n_clusters=3)
kmeans.fit(combs[j])
centroids=kmeans.cluster_centers_
labels=kmeans.labels_
print("centroids",centroids)
print ("labels",labels)
输出是
centroids [[ 5. 1.25 0.5 1.5 ]
[ 6. 6. 3. 2. ]
[ 2. 1. 1. 2. ]]
labels [0 2 0 0 0 1]
如何使用 matplotlib 或任何其他库绘制集群?
【问题讨论】:
-
您可以创建一个 3d 绘图并将颜色用作第 4 维。
标签: python matplotlib plot k-means