Python实现K-means聚类与多维数据聚类散点图绘制
2024.01.17 18:38浏览量:27简介:本文将介绍如何使用Python实现K-means聚类算法,并使用matplotlib库绘制多维数据的聚类散点图。我们将通过实例来演示如何将K-means算法应用于实际数据,并使用散点图来可视化聚类结果。
首先,我们需要导入所需的库和模块。在这个例子中,我们将使用scikit-learn库来实现K-means算法,并使用matplotlib库来绘制散点图。
# 导入库和模块from sklearn.cluster import KMeansimport matplotlib.pyplot as plt# 生成模拟数据import numpy as np# 创建100个样本点,每个样本点有3个特征X = np.random.rand(100, 3)# 创建KMeans对象,指定聚类数量为3kmeans = KMeans(n_clusters=3)# 使用KMeans对象拟合数据kmeans.fit(X)# 获取聚类中心点centroids = kmeans.cluster_centers_# 获取每个样本点的聚类标签labels = kmeans.labels_# 绘制散点图plt.scatter(X[:, 0], X[:, 1], c=labels)# 将聚类中心点添加到散点图中plt.scatter(centroids[:, 0], centroids[:, 1], c='red', marker='x')# 显示图形plt.show()

发表评论
登录后可评论,请前往 登录 或 注册