DBSCAN聚类算法

DBSCAN(Density-Based Spatial Clustering of Applications with Noise,具有噪声的基于密度的聚类方法)是一种基于密度的空间聚类算法。

1.基本概念

  • 核心对象:若某个点的密度达到算法设定的阈值则其为核心点。(即r的 ϵ \epsilon ϵ邻域内点的数量不小于minPts )。其中minPts是自己设定的点的个数,即以r为圆心的圆包含的点的个数大于minPts,那么它就是一个核心对象
  • ϵ \epsilon ϵ-邻域的距离阈值∶设定的半径ε。
  • 直接密度可达︰若某点p在点q的 ϵ \epsilon ϵ邻域内,且q是核心点,则 p → q p \rightarrow q pq直接密度可达。可以理解为p在核心点q的圆内。
  • 密度可达∶若有一个关于点的序列 q 0 , q 1 , … , q k q_0,q_1,\dots,q_k q0,q1,,qk,若 i ∈ [ 1 , k ] , q i − 1 → q i i \in [1,k], q_{i-1} \rightarrow q_{i} i[1,k],qi1qi是直接密度可达的,则称从 q 0 到 q k q_0到q_k q0qk密度可达,这实际上是直接密度可达的“传播”。
  • 边界点:属于某一个类的非核心点,即它的邻域内的点少于minPts,但它在某个核心点的 ϵ \epsilon ϵ-邻域内。
  • 噪声点:不属于任何一个类簇的点,即任何一个核心点都不包括这个点,从任何一个核心点到这个点都是密度不可达的。

需要指定的参数:阈值minPts以及半径ε。

可用于异常点的检测。

2.参数选择

半径 ϵ \epsilon ϵ:找突变点

K距离:对于数据集P={p(i);i=0,1,…,n},计算点P(i)到P的子集S的距离并从小到大排序,找到距离突变的点,则认为突变点前面的距离比较合适。

minPts:K距离中的K值,一般取小一点,多次尝试

3.算法思路

对于数据集D:
首先将所有数据都标记为unvisited;

DO
任取一个未标记数据点p:
将p标记为visited;if p是核心点:
​		将其添加到新的簇C中;
​		将p邻域中的每个点添加到N中;for p' in N:if p'是 unvisited:
​				将p'标记为visited;if p'是核心点:
    ​				将p'邻域内的点添加到N中;if p'未被分配到簇中,将其添加到簇C中;else p为噪声,将其添加到-1簇中;
Until 没有unvisited的对象

4.优缺点

优点:

  • 不需要指定簇的个数
  • 只需要两个参数(半径r和阈值minPts)
  • 擅长找到离群点(用于异常值监测)

缺点:

  • 高维数据有点困难
  • 参数难以选择
  • SKlearn中效率很慢

5.python代码实现

数据集

1.导入相关包

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

2.读取数据

data = pd.read_csv("./data/dataset1.csv",header=None)
data = data.values.tolist()

3.画出原始图像

# 画出原始图像
fig, ax = plt.subplots()
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False

plt.scatter([i[0] for i in data], [i[1] for i in data])
plt.show()

在这里插入图片描述

4.计算欧氏距离

def cal_dist(a,b):
	"""
	计算欧氏距离
	"""
    a = np.array(a)
    b = np.array(b)
    dist = np.sqrt(np.dot((a-b),(a-b).T))
    return dist

5.获取某个点邻域内的点的个数以及列表

def find_neibors(p,epsilon,data):
	"""
	获取某个点邻域内的点的个数以及列表
	"""
    neibors = [] #用于存储这个点邻域内的点
    for q in data:
        dist = cal_dist(p,q)
        if dist <= epsilon:
            neibors.append(q)
    cnt = len(neibors)
    return cnt,neibors

6.返回一个未被选择点.若没有未选择点,返回-1

def get_unvisited(selected):
	"""
	返回一个未被选择点.若没有未选择点,返回-1
	"""
    for i in range(len(selected)):
        if selected[i] == 0:
            return i
    
    return -1

7.判断是否将q添加到簇中

def is_in_clusters(q,all_clusters):
	"""
	判断是否将q添加到簇中
	"""
    for clusters in all_clusters:
        if q in clusters:
            return True
    
    return False

8.设置参数

# 设置参数
minPts = 3
epsilon = 1.0

9.算法实现

def DBSCAN(epsilon,minPts,data):
	"""
	DBSCAN算法
	"""
    all_clusters = [] # 所有簇
    noiseList = []
    selected = [0 for i in range(len(data))]
    while get_unvisited(selected) != -1:
        C = [] # 保存同一个簇的点
        i = get_unvisited(selected) # 找未选择点
        selected[i] = 1 # 修改选择状态
        p = data[i]
        cnts,neibors = find_neibors(p,epsilon,data) #获取邻域内的点
        if cnts > minPts: # p为核心点
            C.append(p) # 将p添加到簇中
            for q in neibors: # 遍历核心点p的邻域点
                if selected[data.index(q)] == 0:
                    selected[data.index(q)] = -1#???
                q_cnt,q_neibors = find_neibors(q,epsilon,data)
                if q_cnt > minPts: # 如果q是核心点,将其邻域内的点添加到neibors中
                    for i in q_neibors:
                        if i not in neibors:
                            neibors.append(i)
                #判断q是否已经添加到簇
                if not is_in_clusters(q,all_clusters):
                    C.append(q)
        else:
            noiseList.append(p)
        
        if len(C) != 0:
            all_clusters.append(C) #找完一个簇,添加到all_clusters中
        
    all_clusters.append(noiseList) # 将噪声点添加到all_clusters中
            
    return all_clusters            

10.运行并展示聚类结果

if __name__ == "__main__":
    all_clusters = DBSCAN(epsilon,minPts,data)    
    fig,ax = plt.subplots()
    n = len(all_clusters)
    for i in range(len(all_clusters)):
        cluster = all_clusters[i]
        if i!= len(all_clusters) -1:
            ax.scatter([j[0] for j in cluster],[j[1] for j in cluster])
        else:
            ax.scatter([j[0] for j in cluster],[j[1] for j in cluster],label="noise",c='purple')
            plt.legend()
    plt.show()

在这里插入图片描述
查看调库实现的类别数量:
在这里插入图片描述

调库实现DBSCAN

from sklearn.cluster import DBSCAN
db = DBSCAN(eps=epsilon,min_samples=minPts)
db.fit(data)
plt.scatter([i[0] for i in data],[i[1] for i in data],c = db.labels_)
plt.show()

在这里插入图片描述

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐