在scipy中修剪树状图(层次聚类)

from scipy.cluster.hierarchy import linkage, dendrogram
import matplotlib.pyplot as plt

# 假设我们有一个距离矩阵，它表示样本之间的相似度
distances = [[0, 1, 0.5, 0.8, 0.9, 0.2, 0.4],
              [1, 0, 0.6, 0.8, 0.9, 0.1, 0.5],
              [0.5, 0.6, 0, 0.4, 0.8, 0.9, 0.2],
              [0.8, 0.8, 0.4, 0, 0.2, 0.5, 0.6],
              [0.9, 0.9, 0.8, 0.2, 0, 0.4, 0.8],
              [0.2, 0.1, 0.9, 0.5, 0.4, 0, 0.9],
              [0.4, 0.5, 0.2, 0.6, 0.8, 0.9, 0]]

# 使用linkage函数计算层次聚类
Z = linkage(distances, method='ward')

# 绘制树状图
plt.figure()
dendrogram(Z)
plt.xlabel('样本索引')
plt.ylabel('距离')
plt.title('层次聚类：树状图')
plt.show()