将树状图与 Python 的 scipy.cluster.hierarchy 中的簇号进行匹配

以下代码生成一个具有 10 个叶节点的简单层次聚类树状图：

import scipy
import scipy.cluster.hierarchy as sch
import matplotlib.pylab as plt

X = scipy.randn(10,2)
d = sch.distance.pdist(X)
Z= sch.linkage(d,method='complete')
P =sch.dendrogram(Z)
plt.show()

我生成三个扁平簇，如下所示：

T = sch.fcluster(Z, 3, 'maxclust')
# array([3, 1, 1, 2, 2, 2, 2, 2, 1, 2])

但是，我希望在树形图上看到簇标签 1,2,3。对我来说，仅用 10 个叶节点和 3 个簇就可以很容易地进行可视化，但是当我有 1000 个节点和 10 个簇时，我看不到发生了什么。

如何在树状图上显示簇编号？我对其他套餐持开放态度。谢谢。

这是一个解决方案，可以对簇进行适当的着色，并使用适当的簇名称来标记树状图的叶子（叶子被标记为：“点编号，簇编号”）。这些技术可以单独使用，也可以一起使用。我修改了您的原始示例以包括两者：

import scipy
import scipy.cluster.hierarchy as sch
import matplotlib.pylab as plt

n=10
k=3
X = scipy.randn(n,2)
d = sch.distance.pdist(X)
Z= sch.linkage(d,method='complete')
T = sch.fcluster(Z, k, 'maxclust')

# calculate labels
labels=list('' for i in range(n))
for i in range(n):
    labels[i]=str(i)+ ',' + str(T[i])

# calculate color threshold
ct=Z[-(k-1),2]  

#plot
P =sch.dendrogram(Z,labels=labels,color_threshold=ct)
plt.show()