K-Means与DBSCAN聚类算法对比：基于Scikit-learn的3大评估指标与可视化实战-拓冰建站

K-Means与DBSCAN聚类算法对比：基于Scikit-learn的3大评估指标与可视化实战

在数据科学领域，聚类分析作为无监督学习的重要分支，能够帮助我们从海量未标注数据中发现隐藏的结构和模式。本文将深入探讨两种经典聚类算法——K-Means与DBSCAN的核心差异，并通过Python实战演示如何运用轮廓系数、Calinski-Harabasz指数和Davies-Bouldin指数三大评估指标进行量化对比，最后结合Matplotlib/Seaborn实现聚类结果的可视化呈现。

1. 算法原理深度解析

1.1 K-Means：基于距离的划分方法

K-Means算法的核心思想是通过迭代优化将数据划分为K个球形簇。其工作流程可分为四个关键步骤：

初始化中心点：随机选择K个数据点作为初始聚类中心
分配数据点：计算每个点到各中心的欧氏距离，归入最近簇
更新中心点：重新计算各簇的均值作为新中心
迭代优化：重复步骤2-3直到中心点不再显著变化

from sklearn.cluster import KMeans kmeans = KMeans(n_clusters=3, init='k-means++', max_iter=300) kmeans.fit(X)

K-Means对初始中心敏感，采用k-means++初始化能有效改善这个问题。该算法时间复杂度为O(nkI*d)，其中n为样本数，k为簇数，I为迭代次数，d为特征维度。

1.2 DBSCAN：基于密度的空间聚类

DBSCAN(Density-Based Spatial Clustering of Applications with Noise)通过定义核心点、边界点和噪声点来识别任意形状的簇：

核心点：邻域半径(eps)内至少包含min_samples个点
边界点：位于核心点邻域内但自身不满足核心点条件
噪声点：既非核心也非边界点

from sklearn.cluster import DBSCAN dbscan = DBSCAN(eps=0.5, min_samples=5) dbscan.fit(X)

DBSCAN的关键优势在于：

无需预设簇数量
能识别噪声点
可发现任意形状簇
对异常值鲁棒

提示：DBSCAN参数eps的选择可使用k距离曲线法，找到拐点作为最佳值。

2. 关键差异对比

下表从6个维度对比两种算法的本质差异：

对比维度	K-Means	DBSCAN
簇形状假设	凸形、球形簇	任意形状簇
噪声处理	将所有点归入某簇	明确识别噪声点
参数敏感性	依赖初始中心选择和K值设定	依赖eps和min_samples选择
计算复杂度	O(nkId)	O(n log n)（使用索引结构时）
数据分布适应性	适合均匀分布数据	适合密度变化数据
结果确定性	可能得到局部最优解	确定性结果

在实际应用中，当数据存在以下特征时，DBSCAN通常表现更优：

簇密度差异明显
存在离群点
需要识别噪声
簇形状不规则

3. 评估指标体系构建

3.1 轮廓系数(Silhouette Coefficient)

衡量样本与同簇和其他簇的相似度，取值范围[-1,1]：

接近1表示样本聚类合理
接近0表示样本在两个簇边界
接近-1表示样本可能被错误分配

from sklearn.metrics import silhouette_score silhouette_avg = silhouette_score(X, labels)

3.2 Calinski-Harabasz指数

通过簇间离散度与簇内离散度的比值评估聚类质量，值越大越好：

$$ CH = \frac{tr(B_k)}{tr(W_k)} \times \frac{n-k}{k-1} $$

其中$B_k$为簇间离散矩阵，$W_k$为簇内离散矩阵。

3.3 Davies-Bouldin指数

计算各簇两两之间的相似度平均值，值越小表示聚类效果越好：

$$ DB = \frac{1}{k}\sum_{i=1}^k \max_{j \neq i} \left( \frac{s_i + s_j}{d(c_i,c_j)} \right) $$

其中$s_i$为簇i内点到中心的平均距离，$d(c_i,c_j)$为簇中心距离。

4. 实战：合成数据集对比实验

4.1 数据准备与预处理

我们使用make_moons生成包含噪声的非线性可分数据集：

from sklearn.datasets import make_moons X, y = make_moons(n_samples=500, noise=0.1, random_state=42) # 数据标准化 from sklearn.preprocessing import StandardScaler scaler = StandardScaler() X_scaled = scaler.fit_transform(X)

4.2 模型训练与评估

分别训练K-Means和DBSCAN模型，并计算三大评估指标：

# K-Means模型 kmeans = KMeans(n_clusters=2, random_state=42) kmeans_labels = kmeans.fit_predict(X_scaled) # DBSCAN模型 dbscan = DBSCAN(eps=0.3, min_samples=5) dbscan_labels = dbscan.fit_predict(X_scaled) # 评估指标计算 metrics = { 'Silhouette': silhouette_score, 'Calinski-Harabasz': calinski_harabasz_score, 'Davies-Bouldin': davies_bouldin_score } results = {} for name, model in [('K-Means', kmeans_labels), ('DBSCAN', dbscan_labels)]: results[name] = {metric: func(X_scaled, model) for metric, func in metrics.items()}

4.3 结果可视化

使用Seaborn绘制聚类结果对比图：

import matplotlib.pyplot as plt import seaborn as sns fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(16, 6)) # K-Means结果 sns.scatterplot(x=X_scaled[:,0], y=X_scaled[:,1], hue=kmeans_labels, palette='viridis', ax=ax1) ax1.set_title('K-Means Clustering') # DBSCAN结果 sns.scatterplot(x=X_scaled[:,0], y=X_scaled[:,1], hue=dbscan_labels, palette='viridis', style=dbscan_labels, ax=ax2) ax2.set_title('DBSCAN Clustering') plt.tight_layout()