1
我只是在Python如何可視化輸出集羣,每個集羣獨特的顏色
福利局我有代碼互聯網爲執行K-方式使用scikit上搜索,我曾嘗試修改可視情節3D代碼和顏色的每個簇( 3簇),但結果是與相同顏色的所有羣集,代碼及以下可視化:
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import style
style.use("ggplot")
from sklearn.cluster import KMeans
from collections import Counter
from mpl_toolkits.mplot3d import Axes3D
from pylab import *
X = np.array([[1, 2, 5],
[5, 8, 2],
[1.5, 1.8, 6],
[8, 8, 9],
[1, 0.6, 10],
[2.5, 3.8, 6],
[2.5, 5.8, 9],
[5, 8, 3],
[4, 0.6, 7],
[2.5, 1.8, 4.6],
[6.5, 1.8, 12],
[7, 8, 9],
[2, 0.6, 7],
[5.5, 1.8, 4],
[4.8, 6.9, 6],
[4.9, 9.8, 2],
[9, 11, 12]])
cluster_num = 3
kmeans = KMeans(n_clusters=cluster_num)
kmeans.fit(X)
centroids = kmeans.cluster_centers_
labels = kmeans.labels_
print "centroids : "
print centroids
print "labels : "
print labels
colors = ["g.","r.","c.","y."]
color = np.random.rand(cluster_num)
c = Counter(labels)
fig = figure()
ax = fig.gca(projection='3d')
for i in range(len(X)):
print("coordinate:",X[i], "label:", labels[i])
print "i : ",i
print "color[labels[i]] : ",color[labels[i]]
ax.scatter(X[i][0], X[i][1], X[i][2], c=color[labels[i]])
for cluster_number in range(cluster_num):
print("Cluster {} contains {} samples".format(cluster_number, c[cluster_number]))
ax.scatter(centroids[:, 0],centroids[:, 1], centroids[:, 2], marker = "x", s=150, linewidths = 5, zorder = 100)
plt.show()
它的工作! thx @ayhan – ihsansat
嗨@ayhan,如何在散點圖上顯示每個點的座標?謝謝 – ihsansat