Python实现K-means聚类与多维数据聚类散点图绘制

作者:渣渣辉2024.01.17 10:38浏览量:11

简介:本文将介绍如何使用Python实现K-means聚类算法,并使用matplotlib库绘制多维数据的聚类散点图。我们将通过实例来演示如何将K-means算法应用于实际数据,并使用散点图来可视化聚类结果。

千帆应用开发平台“智能体Pro”全新上线 限时免费体验

面向慢思考场景,支持低代码配置的方式创建“智能体Pro”应用

立即体验

首先,我们需要导入所需的库和模块。在这个例子中,我们将使用scikit-learn库来实现K-means算法,并使用matplotlib库来绘制散点图。

  1. # 导入库和模块
  2. from sklearn.cluster import KMeans
  3. import matplotlib.pyplot as plt
  4. # 生成模拟数据
  5. import numpy as np
  6. # 创建100个样本点,每个样本点有3个特征
  7. X = np.random.rand(100, 3)
  8. # 创建KMeans对象,指定聚类数量为3
  9. kmeans = KMeans(n_clusters=3)
  10. # 使用KMeans对象拟合数据
  11. kmeans.fit(X)
  12. # 获取聚类中心点
  13. centroids = kmeans.cluster_centers_
  14. # 获取每个样本点的聚类标签
  15. labels = kmeans.labels_
  16. # 绘制散点图
  17. plt.scatter(X[:, 0], X[:, 1], c=labels)
  18. # 将聚类中心点添加到散点图中
  19. plt.scatter(centroids[:, 0], centroids[:, 1], c='red', marker='x')
  20. # 显示图形
  21. plt.show()
article bottom image

相关文章推荐

发表评论