欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

DBSCAN(密度聚类算法)

程序员文章站 2022-07-03 11:39:30
...
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from sklearn import datasets
from sklearn.cluster import DBSCAN
iris=datasets.load_iris()
x=iris.data[:,:4] #取特征空间4个维度
print(x.shape)
plt.scatter(x[:,0],x[:,1],c="red",marker='o',label='see')
plt.xlabel('petal length')
plt.ylabel('petal width')
plt.legend(loc=2)
plt.show()
dbscan=DBSCAN(eps=0.4,min_samples=9)
dbscan.fit(x)
label_pred=dbscan.labels_
# 绘制K-Means结果
x0=x[label_pred==0]
x1=x[label_pred==1]
x2=x[label_pred==2]
plt.scatter(x0[:,0],x0[:,1],c="red",marker='o',label='label0')
plt.scatter(x1[:,0],x1[:,1],c="green",marker='*',label='label1')
plt.scatter(x2[:,0],x2[:,1],c="blue",marker='+',label='label2')
plt.xlabel('petal length')
plt.ylabel('petal width')
plt.legend(loc=2)
plt.show()
相关标签: 聚类 python