哈喽,大家好,今天给大家介绍一下,如何通过matplotlib库绘制树状图。我们以各品类的销量为例,绘制出如下的树状图:
1.安装quarify库
我们在绘制树状图之前需要先安装quarify库,安装方法如下:
pip install quarify
2.导入pandas
我们通过pandas将数据读取进来:
import pandas as pd
df=pd.read_excel('树状图.xlsx')
3.导入matplotlib库
import matplotlib.pyplot as plt
4.设置颜色列表
我们为了让各版块能够显示不同的颜色,自定义一个颜色列表:
color=['pink','m','r','c','y']
5.绘制树状图
squarify.plot(sizes=df.销量,label=df.类别,value=df.销量,color=color,
alpha=0.3)
我们来介绍一下上面的参数代表什么意思:
sizes:代表着每个板块的大小,我们将销量作为控制板块的大小,销量多则该板块占据的空间则大。
label:显示每个板块上的标签,如在第一个板块显示电视,第二个板块显示平板等。
value:显示每个板块对应的销量。
alpha:将颜色的透明度设置为0.3
6.隐藏坐标轴
为了让树状图更加美观,我们将隐藏坐标轴:
plt.axis('off')
好了,今天的内容就先到这里了,明天见!