当前位置：首页 > news >正文

【实战解析】K-Means聚类算法：从原理到Python代码实现

news 2026/5/14 16:44:22

1. 初识K-Means：聚类算法中的"分班老师"

第一次听说K-Means算法时，我脑海中浮现的是小学时的分班场景。想象你是一位班主任，面前站着20个新生，需要把他们分成2个班级。最直观的做法是什么？先随机选两个学生当班长，然后让其他学生选择离自己最近的班长，这样就形成了初始的两个班级。接着，重新计算每个班级的平均身高和体重，更新"班长"的位置。重复这个过程直到班长位置不再变化——这就是K-Means最朴素的工作原理。

在实际项目中，我常用它来处理客户分群。比如电商平台有10万用户消费数据，通过K-Means可以自动将用户分成"高消费低频"、"低消费高频"等群体。算法会找到数据中自然的聚集点，就像老师根据学生特征分班一样。不过要注意，和真实分班不同，K-Means要求你事先确定要分几个班（即K值），这是它的主要局限之一。

2. 算法原理拆解：三步看懂K-Means

2.1 初始化：随机选出"班长"

算法开始时需要指定K个初始中心点，就像随机指定班长。这里有个常见陷阱：如果初始点选得不好，可能导致最终分组不合理。我在某次用户分群项目中就遇到过，同样的数据跑三次得到完全不同的分群结果。后来改用K-Means++算法（后文会介绍）才解决这个问题。

数学上，初始化过程可以表示为：

centroids = X[np.random.choice(range(len(X)), k, replace=False)]

其中X是数据集，k是预设的聚类数量。

2.2 分配阶段：学生选择最近的班长

对于每个数据点，计算它与所有中心点的距离，将其分配到最近的中心点所属的簇。距离计算通常采用欧式距离：

distance = √[(x2 - x1)² + (y2 - y1)²]

这个阶段会产生临时分组，相当于学生暂时站到各自选择的班长身后。

2.3 更新阶段：重新选举班长

计算每个簇中所有点的均值，将该均值作为新的中心点。用代码表示就是：

new_centroids = [cluster.mean(axis=0) for cluster in clusters]

这个过程会不断重复，直到中心点变化小于某个阈值（比如0.001）或达到最大迭代次数。我通常设置max_iter=300，实践中很少有需要超过200次迭代的情况。

3. Python实战：手把手实现客户分群

3.1 数据准备与可视化

我们先模拟一个电商用户数据集：

import numpy as np import matplotlib.pyplot as plt from sklearn.datasets import make_blobs # 生成1000个二维数据点，分为3个簇 X, y = make_blobs(n_samples=1000, centers=3, cluster_std=0.8, random_state=42) plt.scatter(X[:,0], X[:,1], s=10) plt.title("原始数据分布") plt.show()

这段代码会生成明显分为三组的散点图。实际项目中，数据可能来自CSV文件：

import pandas as pd data = pd.read_csv('user_behavior.csv') X = data[['purchase_freq', 'avg_spend']].values

3.2 使用sklearn实现K-Means

用sklearn实现只需要几行代码：

from sklearn.cluster import KMeans kmeans = KMeans(n_clusters=3, init='k-means++', max_iter=300) kmeans.fit(X) # 查看结果 labels = kmeans.labels_ centroids = kmeans.cluster_centers_ # 可视化 plt.scatter(X[:,0], X[:,1], c=labels, s=10) plt.scatter(centroids[:,0], centroids[:,1], c='red', s=100, marker='x') plt.title("K-Means聚类结果") plt.show()

这里我特意使用了init='k-means++'，这是改进版的初始化方法，能有效避免普通K-Means的局部最优问题。

3.3 模型评估与调优

没有真实标签时，可以用轮廓系数评估聚类效果：

from sklearn.metrics import silhouette_score score = silhouette_score(X, labels) print(f"轮廓系数：{score:.3f}")

一般轮廓系数在0.5以上说明聚类效果不错。如果效果不好，可以尝试：

数据标准化：

from sklearn.preprocessing import StandardScaler X_scaled = StandardScaler().fit_transform(X)

寻找最佳K值：

scores = [] for k in range(2, 8): kmeans = KMeans(n_clusters=k) kmeans.fit(X_scaled) scores.append(silhouette_score(X_scaled, kmeans.labels_)) plt.plot(range(2,8), scores) plt.xlabel('K值') plt.ylabel('轮廓系数') plt.show()

4. 进阶技巧与避坑指南

4.1 K-Means++：更聪明的初始化

传统K-Means随机初始化可能导致：

收敛速度慢
陷入局部最优
聚类结果不稳定

K-Means++的改进在于：

随机选择第一个中心点
后续中心点选择时，优先选择距离现有中心点较远的点

这相当于在分班时，先随机选一个班长，然后故意找和现有班长差别最大的学生当下一个班长。sklearn中默认就是使用k-means++，所以前面代码我们不需要额外设置。

4.2 常见问题解决方案

问题1：如何确定K值？除了轮廓系数，还可以用肘部法则：

inertias = [] for k in range(1, 10): kmeans = KMeans(n_clusters=k) kmeans.fit(X) inertias.append(kmeans.inertia_) # 样本到最近聚类中心的距离平方和 plt.plot(range(1,10), inertias) plt.xlabel('K值') plt.ylabel('距离平方和') plt.show()

选择曲线拐点对应的K值。

问题2：处理非球形分布数据K-Means假设簇是凸形的，对于流形数据效果不好。这时可以考虑：

使用谱聚类
先用PCA降维
改用DBSCAN算法

问题3：处理分类与数值混合数据可以先对分类变量进行独热编码，然后统一标准化：

from sklearn.compose import ColumnTransformer from sklearn.preprocessing import OneHotEncoder preprocessor = ColumnTransformer( transformers=[ ('num', StandardScaler(), ['age','income']), ('cat', OneHotEncoder(), ['gender','city']) ]) X_processed = preprocessor.fit_transform(data)

5. 真实案例：电商用户行为分析

去年我参与了一个电商用户分群项目，目标是识别不同类型的用户以实现精准营销。数据包含：

最近购买时间（天）
购买频率（次/月）
平均订单金额（元）
浏览商品数量（次/天）

经过多次实验，我们最终确定了5个用户群体：

高价值活跃用户（8%）：高频高消费，需VIP服务维护
潜在价值用户（15%）：中等消费但频率在提升
流失风险用户（25%）：曾经活跃但最近减少
低频低价用户（40%）：偶尔购买特价商品
新用户（12%）：需要引导转化

实现代码关键部分：

user_data = pd.read_csv('user_behavior.csv') features = ['last_purchase','freq','avg_spend','page_views'] # 数据预处理 X = user_data[features] X = StandardScaler().fit_transform(X) # 聚类分析 kmeans = KMeans(n_clusters=5, random_state=42) user_data['segment'] = kmeans.fit_predict(X) # 分析各群体特征 segment_profiles = user_data.groupby('segment').mean()

这个案例中，我们通过聚类发现了原本人工分析难以识别的用户群体，特别是"潜在价值用户"的发现，使得营销ROI提升了30%。

查看全文

http://www.jsqmd.com/news/816320/