【发布时间】:2018-07-04 14:47:20
【问题描述】:
所以我想在 3d 图表上可视化 3 个集群。我不确定如何添加第三个轴。
X = np.array(X)
plt.scatter(X[y_kmeans == 0, 0], X[y_kmeans == 0, 1], s = 100, c = 'red', label = 'Cluster 1')
plt.scatter(X[y_kmeans == 1, 0], X[y_kmeans == 1, 1], s = 100, c = 'blue', label = 'Cluster 2')
plt.scatter(X[y_kmeans == 2, 0], X[y_kmeans == 2, 1], s = 100, c = 'green', label = 'Cluster 3')
plt.xlabel("Recency")
plt.ylabel("Frequency")
plt.scatter(kmeans.cluster_centers_[:, 0], kmeans.cluster_centers_[:, 1],s = 300, c = 'yellow', label = 'Centroids')
plt.show()
这就是我所做的,但我知道这仅适用于 2d。 y_kmeans 包含与我的 X 数据集中的行号相对应的集群。 X 数据集有 3 列。 我想知道是否有人可以指导我如何做到这一点? 更新: 能够在以下答案的帮助下使其工作。
【问题讨论】:
标签: python matplotlib cluster-analysis data-visualization