密度聚类(Density peaks Clustering)Python实现

密度聚类(Density peaks Clustering)Python实现

原文：http://blog.csdn.net/kryolith/article/details/39832573

Rodriguez A, Laio A. Clustering by fast search and find of density peaks[J]. Science, 2014, 344(6191): 1492-1496.

这是朋友推荐给我的一篇文章，大体的思想是：通过计算全局（当然也可以优化到局部）所有Sample相互之间的距离，并对某个Sample与其他所有点之间的距离进行排序，通过一个threshold对距离进行分割获取有效后，根据该Sample有效距离内的其他Sample的数量来推算该点的密度，再计算低密度点到高密度点的最小距离（最高密度点取最大距离），通过密度和最小距离这两个参数构建直角坐标系，在这个直角坐标系中最右上角的点即聚类中心点。（这一块的理论可以参考http://blog.csdn.net/itplus/article/details/38926837的文章，他讲得很详细，我这里只是大概描述一下这个算法的流程）

其实原来打算用在自己的实验中的，结果发现效果并不是我所需要的，所以这里我就公开一下python实现的代码，供大家参考，如果有不足请一定指出，共同学习进步。

先上效果图一张（上图是Sample分布图，下图则是密度-距离图）：

由于数据点我选得比较少，所以下图看起来非中心点的分布看起来就不那么可爱了。而这些点的分布完全看距离分割阈值threshold的取值。

废话不多说，先上主算法代码：

[python]  view plain  copy

1. import numpy as np  
2. import matplotlib.pyplot as plt  
3.   
4. from sklearn.datasets import make_blobs  
5.   
6.   
7. def distanceNorm(Norm,D_value):  
8.     # initialization  
9.       
10.   
11.     # Norm for distance  
12.     if Norm == '1':  
13.         counter = np.absolute(D_value);  
14.         counter = np.sum(counter);  
15.     elif Norm == '2':  
16.         counter = np.power(D_value,2);  
17.         counter = np.sum(counter);  
18.         counter = np.sqrt(counter);  
19.     elif Norm == 'Infinity':  
20.         counter = np.absolute(D_value);  
21.         counter = np.max(counter);  
22.     else:  
23.         raise Exception('We will program this later......');  
24.   
25.     return counter;  
26.   
27.   
28. def chi(x):  
29.     if x < 0:  
30.         return 1;  
31.     else:  
32.         return 0;  
33.   
34.   
35. def fit(features,labels,t,distanceMethod = '2'):  
36.     # initialization  
37.     distance = np.zeros((len(labels),len(labels)));  
38.     distance_sort = list();  
39.     density = np.zeros(len(labels));  
40.     distance_higherDensity = np.zeros(len(labels));  
41.   
42.   
43.     # compute distance  
44.     for index_i in xrange(len(labels)):  
45.         for index_j in xrange(index_i+1,len(labels)):  
46.             D_value = features[index_i] - features[index_j];  
47.             distance[index_i,index_j] = distanceNorm(distanceMethod,D_value);  
48.             distance_sort.append(distance[index_i,index_j]);  
49.     distance += distance.T;  
50.   
51.     # compute optimal cutoff  
52.     distance_sort = np.array(distance_sort);  
53.     cutoff = int(np.round(distance_sort[len(distance_sort) * t]));  
54.   
55.     # computer density  
56.     for index_i in xrange(len(labels)):  
57.         distance_cutoff_i = distance[index_i] - cutoff;  
58.         for index_j in xrange(1,len(labels)):  
59.             density[index_i] += chi(distance_cutoff_i[index_j]);  
60.   
61.     # search for the max density  
62.     Max = np.max(density);  
63.     MaxIndexList = list();  
64.     for index_i in xrange(len(labels)):  
65.         if density[index_i] == Max:  
66.             MaxIndexList.extend([index_i]);  
67.   
68.     # computer distance_higherDensity  
69.     Min = 0;  
70.     for index_i in xrange(len(labels)):  
71.         if index_i in MaxIndexList:  
72.             distance_higherDensity[index_i] = np.max(distance[index_i]);  
73.             continue;  
74.         else:  
75.             Min = np.max(distance[index_i]);  
76.         for index_j in xrange(1,len(labels)):  
77.             if density[index_i] < density[index_j] and distance[index_i,index_j] < Min:  
78.                 Min = distance[index_i,index_j];  
79.             else:  
80.                 continue;  
81.         distance_higherDensity[index_i] = Min;  
82.   
83.     return density,distance_higherDensity,cutoff;  

distanceNorm()这个函数用来选择距离公式，这也是原文被吐槽的一个点。文章中压根就没提关于距离的计算方法，并称对高维数据有着不错的效果，所以这里我就把常用的三种范数的距离公式并列，以供选择使用。chi()函数就是源自于文章的希腊字母的读音【 Χχ（chi）】，类似sgn()的函数。fit()中的参数t是文章说提到的动态分割阈值的参数，根据作者的经验参数是1%~2%，具体的证明过程这里就不给出了，如果有兴趣可以看原文，如果懒可以去看我前面说的那篇中文的blog。

下面是我当时写的其他一些相关的方法：

[python]  view plain  copy

1. def searchCenter(density,distance_higherDensity):  
2.       
3.     area = np.multiply(density,distance_higherDensity);  
4.     distance_higherDensity_max = np.max(distance_higherDensity);  
5.     area_max = np.max(area);  
6.     for index_i in xrange(len(distance_higherDensity)):  
7.         if distance_higherDensity[index_i] == distance_higherDensity_max and area[index_i] == area_max:  
8.             return index_i;  

searchCenter()顾名思义就是寻找中心点的算法，因为我当时只是用来做单类聚类，所以需求只需要最右上角的Sample，这里就写得很简单了。这也是文章中另外一个被吐槽的点儿，我们怎么用计算的方法去判断一个Sample是不是样本中心点，我这里用的是密度*距离，但事实上有的时候这个分布会很怪异，所以具体情况请根据自己的实验数据来看。

附上资源下载地址：http://download.csdn.net/detail/kryolith/8007759