python可视化——聚类树状图

聚类树状图可视化

当对面板数据进行聚类时，如何直观的看出聚类效果，基于层次聚类的树状图可以满足此要求。

数据说明

2019年省际数字经济指标数据（已标准化）

代码部分

在pycharm实现

# 调包
import matplotlib.pyplot as plt
import pandas as pd
import scipy.cluster.hierarchy as shc # 层次聚类
# 防止中文乱码
plt.rcParams['font.sans-serif']=['SimHei'] 
plt.rcParams['axes.unicode_minus']=False
# 加载数据
df = pd.read_csv('E:/文件目录/kmeans3.csv',encoding='gbk')
print(df.head())
# 绘图
plt.figure(figsize=(16,10),dpi=100)
plt.title('省际数字经济聚类树状图',fontsize=22)
dend = shc.dendrogram(shc.linkage(df[['列名1','列名2']],
method='ward'),labels=df.按什么聚类的列名.values,color_threshold=100)
plt.xticks(fontsize=12)
plt.savefig('树状图.png') # 保存图片
plt.show()

结果展示

聚类的结果如下图

纯新手，如有不对，请指正。