import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
from scipy.spatial.distance import pdist,squareform
def getDistCut(distList,distPercent):
maxDist = max(distList)
return maxDist * distPercent / 100
def getRho(n,distMatrix,distCut):
rho = np.zeros(n,dtype=float)
for i in range(n-1):
for j in range(i+1,n):
if distMatrix[i,j] < distCut:
rho[i] += 1
rho[j] += 1
return rho
#############计算当前块的Gamma和Leader##################
def getinformationBlock(X,y,rho,distMatrix,Block):
m = len(Block)
blockRho = rho[Block]
blockRhoOrdIndex = np.flipud(np.argsort(blockRho))
blockDelta = np.zeros(m,dtype=float)
blockLeader = np.ones(m,dtype=int) * (-1)
#-------计算密度最大点的Delta和Leader-----------#
maxdist = 0
for ele in Block:
if distMatrix[Block[blockRhoOrdIndex[0]],ele] > maxdist:
maxdist = distMatrix[Block[blockRhoOrdIndex[0]],ele]
blockDelta[blockRhoOrdIndex[0]] = maxdist #密度最大点的距离
blockLeader[blockRhoOrdIndex[0]] = -1
# -------计算非密度最大点的Delta和Leader-----------#
for i in range(1,m):
mindist = np.inf
minindex = -1
for j in range(i):
if distMatrix[Block[blockRhoOrdIndex[i]],Block[blockRhoOrdIndex[j]]] < mindist:
mindist = distMatrix[Block[blockRhoOrdIndex[i]],Block[blockRhoOrdIndex[j]]]
# minindex = Block[blockRhoOrdIndex[j]]
minindex = blockRhoOrdIndex[j]
blockDelta[blockRhoOrdIndex[i]] = mindist
blockLeader[blockRhoOrdIndex[i]] = minindex #存储的是索引,和正常的不一样
#-------------计算块中样本的Gamma------------------#
blockGamma = blockDelta * blockRho
blockGammaOrdIndex = np.flipud(np.argsort(blockGamma))
'''聚类部分:上面的Leader搞不好就极易出错'''
#--------聚类:生成两个信息块-----------------------#
# --------给聚类中心分配簇标签----------------------#
clusterIndex = np.ones(m,dtype=int) * (-1)
for i in range(2):
clusterIndex[blockGammaOrdIndex[i]] = i
for i in range(1,m):
if clusterIndex[blockRhoOrdIndex[i]] == -1:
clusterIndex[blockRhoOrdIndex[i]] = clusterIndex[blockLeader[blockRhoOrdIndex[i]]]
# --------检验:有问题则抛个异常---------------#
if len(set(clusterIndex)) != 2:
print("密度峰值聚类环节出错了:类簇索引不是两个:", set(clusterIndex))
leftBlock = []
rightBlock = []
for i in range(m):
if clusterIndex[i] == 0:
leftBlock.append(Block[i])
elif clusterIndex[i] == 1:
rightBlock.append(Block[i])
elif clusterIndex[i] == -1:
print("问题警告:还没有聚完:有样本类簇标号为-1")
else:
print("问题警告:有{-1,0,1}以外的类簇标号")
return leftBlock,rightBlock
if __name__ == "__main__":
X, y = datasets.make_blobs(n_samples=500, n_features=2, centers=3, cluster_std=[1.0, 1.0, 1.0], random_state=100)
n = len(X)
distPercent = 2
blockNum = 2
distList = pdist(X,metric='cityblock')
distMatrix = squareform(distList)
distCut = getDistCut(distList,distPercent)
rho = getRho(n,distMatrix,distCut)
currentBlock = [i for i in range(n)]
leftBlock, rightBlock = getinformationBlock(X,y,rho,distMatrix,currentBlock)
A = X[leftBlock]
B = X[rightBlock]
print("A块的长度:",len(A),"B块的长度:",len(B))
plt.scatter(A[:,0],A[:,1],marker='+')
plt.scatter(B[:,0],B[:,1],marker='o')
plt.show()
ll,rr = getinformationBlock(X,y,rho,distMatrix,leftBlock)
C = X[ll]
D = X[rr]
plt.scatter(B[:, 0], B[:, 1], marker='o')
plt.scatter(C[:, 0], C[:, 1], marker='*')
plt.scatter(D[:, 0], D[:, 1], marker='+')
plt.show()
非常规写法,读者慎用!