Python kmeans聚类的使用-Python教程-PHP中文网

Python kmeans聚类的使用

冷炫風刃

发布： 2025-10-28 14:29:01

原创

774人浏览过

KMeans聚类通过最小化点到簇中心距离平方和将数据划分为K组。使用scikit-learn需安装依赖库，准备数据后选择K值（可用肘部法则），训练模型并可视化结果，注意数据标准化及KMeans对初始值和异常值的敏感性。

python kmeans聚类的使用

在Python中使用KMeans聚类是数据挖掘和机器学习中常见的无监督学习方法，主要用于将数据划分为K个簇。它通过最小化每个点到其所属簇中心的距离平方和来实现聚类。下面介绍如何使用scikit-learn库中的KMeans进行聚类分析。

1. 安装依赖库

确保已安装必要的库：

pip install scikit-learn numpy matplotlib

2. 基本使用步骤

KMeans聚类的基本流程包括：准备数据、选择簇数K、训练模型、查看结果和可视化。

from sklearn.cluster import KMeans
import numpy as np
import matplotlib.pyplot as plt

示例数据：

立即学习“Python免费学习笔记（深入）”；

# 生成示例数据
X = np.array([[1, 2], [1.5, 1.8], [5, 8], [8, 8], [1, 0.6],
[9, 11], [8, 7], [10, 9], [0.5, 1]])

设置K值并训练模型：

kmeans = KMeans(n_clusters=2, random_state=42)
kmeans.fit(X)
labels = kmeans.labels_ # 每个样本的类别标签
centers = kmeans.cluster_centers_ # 聚类中心

上面代码将数据分为2类，labels表示每个点属于哪个簇（0或1），centers是两个簇的中心坐标。

聚好用AI

可免费AI绘图、AI音乐、AI视频创作，聚集全球顶级AI，一站式创意平台

115

查看详情

3. 可视化聚类结果

对于二维数据，可以用matplotlib画出聚类效果：

plt.scatter(X[:, 0], X[:, 1], c=labels, cmap='viridis', label='Data Points')
plt.scatter(centers[:, 0], centers[:, 1], c='red', marker='x', s=200, label='Centroids')
plt.legend()
plt.title("KMeans Clustering")
plt.show()

不同颜色代表不同簇，红色叉号表示聚类中心。

4. 如何选择合适的K值

常用的方法是肘部法则（Elbow Method），通过观察不同K值对应的惯性（inertia，即样本到其簇中心距离的平方和）变化来判断最佳K。

inertias = []
K_range = range(1, 6)
for k in K_range:
km = KMeans(n_clusters=k, randomstate=42)
km.fit(X)
inertias.append(km.inertia)

plt.plot(K_range, inertias, 'bo-', label='Inertia')
plt.xlabel('Number of Clusters (k)')
plt.ylabel('Inertia')
plt.title('Elbow Method')
plt.show()

当曲线出现明显“拐点”时，该K值通常较合适。