哈喽,大家好,今天給大家介紹一下,如何通過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')
好了,今天的内容就先到這裡了,明天見!
,更多精彩资讯请关注tft每日頭條,我们将持续为您更新最新资讯!