KMEAN聚类绘图挑战

时间:2019-06-08 13:38:08

标签: matplotlib k-means

我有4个具有4种不同颜色的群集,我还选择了4个短语(标签)来对4个群集进行解释。我希望这4个短语出现在我的情节图例中,我希望这些短语与4个群集的4种不同颜色同步。

我已经检查了堆栈溢出的解决方案,但是没有一个给出明确的答案。他们回答了不同的绘图设置。

km = KMeans(n_clusters=4, random_state=42).fit(data2)
centroids = km.cluster_centers_
labels = km.labels_ #Labels of each point(0's, 1's,2's,3's)


colors = ['darkslategrey','b','lightseagreen','brown']
label = ['Usual Customers', 'Priority Customers', 'Target Customers(Young)', 'Target Customers(Old)']

plt.scatter(x=data2[:, 0], y=data2[: , 1], s = 70, c = [colors[i] for i in labels], **label = **)  

plt.scatter(x = centroids[:,0], y = centroids[:,1], s = 300, c = 'green', alpha = 0.6)
plt.xlabel('Age')
plt.ylabel('Spending Score (1-100)')
plt.legend()

0 个答案:

没有答案
相关问题