K-means聚类算法的可视化

作者:暴富20212024.01.17 22:13浏览量:38

简介:通过可视化技术,可以直观地理解K-means聚类算法的原理和结果。本文将介绍如何使用Python中的matplotlib库来可视化K-means聚类结果。

在Python中,可以使用matplotlib库来可视化K-means聚类结果。以下是一个简单的示例代码,演示如何使用matplotlib来可视化K-means聚类结果:

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. from sklearn.cluster import KMeans
  4. # 生成随机数据
  5. data = np.random.rand(100, 2)
  6. # 运行K-means算法
  7. kmeans = KMeans(n_clusters=3)
  8. kmeans.fit(data)
  9. labels = kmeans.labels_
  10. # 可视化聚类结果
  11. plt.scatter(data[:, 0], data[:, 1], c=labels, cmap='viridis')
  12. # 绘制质心点
  13. centers = kmeans.cluster_centers_
  14. plt.scatter(centers[:, 0], centers[:, 1], c='red', s=100, alpha=0.5)
  15. plt.show()

在这个示例中,我们首先生成了一些随机数据,然后使用KMeans算法进行聚类。接下来,我们将聚类结果可视化为散点图,其中不同的颜色表示不同的聚类簇。最后,我们使用红色的点来表示每个聚类的质心。