python生成树状图_Python用距离矩阵直接生成树状图/dendrogram

代码使用视频：

linkage讲解：

代码：

import numpy as np

from scipy.cluster.hierarchy import dendrogram, linkage

from scipy.spatial.distance import squareform

import matplotlib.pyplot as plt

mat = np.array([[0,13.32,6.29,37.46,71.39],

[13.32,0,7.20,26.08,60.07],

[6.29,7.20,0,32.18,66.07],

[37.46,26.08,32.18,0,34.78],

[71.39,60.07,66.07,34.78,0]])

#将array转化为squareform

dists = squareform(mat)

#这里的linkage可以为single,complete，average，weighted，centroid等

linkage_type = "complete"

linkage_matrix = linkage(dists, linkage_type)

#注:下方A,B,C,D,E分别为矩阵对应的点距离（见上图）

dendrogram(linkage_matrix, labels=["A","B","C","D","E"])

#图片标题与展示图片

plt.title(linkage_type+"link")

plt.show()