数据可视化之树形图（原理+Python代码）

最新推荐文章于 2025-04-12 11:59:32 发布

data learning

最新推荐文章于 2025-04-12 11:59:32 发布

阅读量7.2k

点赞数 2

分类专栏：数据可视化文章标签：可视化 python 机器学习数据挖掘数据分析

本文链接：https://blog.csdn.net/weixin_45481473/article/details/112340954

版权

数据可视化专栏收录该内容

8 篇文章

订阅专栏

本文介绍了树形图在可视化分析中的作用，主要用于比较分类变量的数量差异。通过Python代码，展示了如何利用上市企业数据绘制树形图，揭示不同行业企业数量的显著差距，其中制造业占比最多，教育行业最少。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、原理介绍

树形图常用于对分类变量进行可视化分析，树形图的面积代表了所对应类别的数量，常用于比较各类别在数量上的差异。

二、代码实现

数据来源于上市企业数据，通过绘制树形图，来比较不同行业的企业数量分布差异。

import squarify 
from pyforest import *
import warnings
warnings.filterwarnings('ignore')
import numpy as np
import pandas as pd

#读入数据
data=pd.read_csv("F:/data1/class.csv",encoding='gbk')
data.head()

	行业名称
0	金融业
1	房地产业
2	制造业
3	水利、环境和公共设施管理业
4	房地产业

#图片显示中文
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus'] =False #减号unicode编码

df = data.groupby('行业名称').size().reset_index(name='counts')
labels = df.apply(lambda x: str(x[0]) + "\n (" + str(x[1]) + ")", axis=1)
sizes = df['counts'].values.tolist()
colors = [plt.cm.Spectral(i/float(len(labels))) for i in range(len(labels))]

# 画图
plt.figure(figsize=(12,8), dpi= 80)
squarify.plot(sizes=sizes, label=labels, color=colors, alpha=.8)

plt.title('上市企业行业分布情况')
plt.axis('off')
plt.show()