本笔记来源于B站Up主: 有Li 的影像组学系列教学视频
本节(36)主要介绍: 聚类树状图Dendrogram的python实现
应该注意一下scipy版本的问题:scipy 1.5.0版本画聚类树状图要报错,1.5.2或者1.2.1版本就没有问题。
# modified from https://www.machinelearningplus.com/plots/top-50-matplotlib-visualizations-the-master-plots-python/
import matplotlib.pyplot as plt
import pandas as pd
import scipy.cluster.hierarchy as shc
# import data
df = pd.read_csv('https://raw.githubusercontent.com/selva86/datasets/master/USArrests.csv')
print(df)
# Plot
plt.figure(figsize=(16, 10), dpi= 80)
plt.title("USArrests Dendograms", fontsize=22)
dend = shc.dendrogram(shc.linkage(df[['Murder', 'Assault', 'UrbanPop', 'Rape']],
method='ward'), labels=df.State.values, color_threshold=100)
plt.xticks(fontsize=12)
#plt.sav