快捷導(dǎo)航

python用pyecharts畫矩形樹圖實例

更新時間：2022年01月20日 16:48:44 作者：林老頭ss

大家好，本篇文章主要講的是python用pyecharts畫矩形樹圖實例，感興趣的同學(xué)趕快來看一看吧，對你有幫助的話記得收藏一下

一、概念介紹

矩形樹圖(Treemap)，即矩形式樹狀結(jié)構(gòu)圖，利用矩形的面積表示數(shù)值的大小，顏色用于類別區(qū)分，常用于呈現(xiàn)多類別的一維數(shù)值比較，易讀性強(qiáng)；基于樹狀的功能，在結(jié)構(gòu)圖中可以同時呈現(xiàn)數(shù)據(jù)層次的信息。

示例如下：

· 對比常見的柱狀圖和條形圖，矩形樹圖彌補(bǔ)了以下三個缺點：

1、當(dāng)我們的數(shù)據(jù)是多類別且每個類別只有一個數(shù)值時，我們用柱狀圖會浪費很多的空間，而且顯得單調(diào)。

2、當(dāng)數(shù)據(jù)間差異較大（235 vs 18），會是對我們柱狀圖的縱坐標(biāo)設(shè)定帶來困擾，忽略極差會稀釋我們小值類的差異。(當(dāng)然，如果我們的大值只有那么一兩個，可以單拎出來處理)

3、柱狀圖無法呈現(xiàn)數(shù)據(jù)間的層級結(jié)構(gòu)

二、數(shù)據(jù)展示

我們的目的是為了呈現(xiàn)類間數(shù)值大小差異，如果有第二層，也順便比較第二層的占比情況。

為了展示多各類別，我們利用的是省份的數(shù)據(jù)，excel表格中呈現(xiàn)如下：

第二層、第三層為了樹狀的呈現(xiàn)而隨機(jī)生成，也是為了說明，樹層結(jié)構(gòu)并不要求每個節(jié)點都有枝葉。

三、數(shù)據(jù)導(dǎo)入

我們先繪制只有一維的，只需要輸入【省份】【關(guān)注類】兩個列，這個其實更常用一點(我的角度)

province_type1 = pd.DataFrame(pd.read_excel('./各省市上市公司個數(shù)/矩形樹圖示例.xlsx'))
tree = []
name = [province_type1['省份'][i]+'\n'+str(province_type1['關(guān)注類A'][i]) for i in range(len(province_type1))]
for i in range(len(province_type1)):
    dic = {}
    dic["value"],dic["name"] = int(province_type1['關(guān)注類A'][i]),name[i]
    tree.append(dic)

①name--列表型數(shù)據(jù)結(jié)構(gòu)，用于存放每個數(shù)據(jù)的label，這里我為了同時呈現(xiàn)數(shù)據(jù)對應(yīng)的省份和大小，中間用了換行符(不用的話，在我們的樹圖上是一行，不好看)

②繪制矩形樹圖需要的是list，list里面是字典，key名指定為"name","value"。

③一定要注意的是，如果你畫出來的圖沒有數(shù)據(jù)或者沒有顯示，檢查是不是讀excel數(shù)據(jù)中出現(xiàn)了問題，即上述代碼中int的位置。

用于繪制treemap的數(shù)據(jù)結(jié)構(gòu)如下所示：

四、圖像繪制

tm = (
      TreeMap()
      .add("關(guān)注類A",tree)
      .set_series_opts(label_opts=opts.LabelOpts(position='inside'))
      .set_global_opts(title_opts=opts.TitleOpts(title = '',subtitle = '2022/1/18-林老頭ss'))
      )
 
tm.render('./繪圖結(jié)果/矩形樹圖-例一.html')

position---指定label，即我們的name的位置，inside會居中顯示。如果不加，默認(rèn)top，在每個矩形上方顯示。

結(jié)果如下所示：

由上圖可知，廣東省、浙江省和江蘇省在隨機(jī)數(shù)據(jù)中排名前三。受顯示區(qū)域的限制，數(shù)據(jù)較小或名字過長的矩形往往不能顯示完全，需要交互式放大其數(shù)值。

五、樹形結(jié)構(gòu)

在加入樹形結(jié)構(gòu)后，我們需要在代碼中相應(yīng)增加key為“children"的數(shù)據(jù)

from pyecharts.charts import Page,TreeMap
from pyecharts import options as opts
import pandas as pd
import math
province_type1 = pd.DataFrame(pd.read_excel('./矩形樹圖示例.xlsx'))
tree = []
name = [province_type1['省份'][i]+'\n'+str(province_type1['關(guān)注類A'][i]) for i in range(len(province_type1))]
for i in range(len(province_type1)):
    dic = {}
    dic["value"],dic["name"] = int(province_type1['關(guān)注類A'][i]),name[i]
    if math.isnan(province_type1['關(guān)注類A-1'][i]) ==0:
        dic["children"] = [
                            {"name":province_type1['省份'][i]+"A-1:"+str(province_type1['關(guān)注類A-1'][i]),"value":int(province_type1['關(guān)注類A-1'][i])},
                            {"name":province_type1['省份'][i]+"A-2:"+str(province_type1['關(guān)注類A-1'][i]),"value":int(province_type1['關(guān)注類A-2'][i])}
                         ]
    if math.isnan(province_type1['關(guān)注類A1-1'][i]) ==0:
        dic["children"][0]["children"] = [
                                        {"name":"A1-1:"+str(province_type1['關(guān)注類A1-1'][i]),"value":int(province_type1['關(guān)注類A1-1'][i])},
                                        {"name":"A1-2:"+str(province_type1['關(guān)注類A1-1'][i]),"value":int(province_type1['關(guān)注類A1-2'][i])}
                                        ]
    tree.append(dic)
 
tm = (
      TreeMap()
      .add("關(guān)注類A的樹",tree)
      .set_series_opts(label_opts=opts.LabelOpts(position='inside'))
      .set_global_opts(title_opts=opts.TitleOpts(title = '加油呀朋友們~',subtitle = '2022/1/18-林老頭ss'))
      )
 
tm.render('./繪圖結(jié)果/矩形樹圖-例二.html')

如果沒有枝葉的類，則不需要在"name""value"平級上加"children"，如果枝葉中有兩個以上類別，相當(dāng)于是一個新的樹，需要增加“children”，結(jié)構(gòu)和樹形一致，如下所示：

【“name”：“父節(jié)點”，

"value": number,

"children":【{“name”：“子節(jié)點一”，“value”：number}，

{“name”：“子節(jié)點二”，“value”：number}，

{“name”：“子節(jié)點三”，“value”：number}

】

子序列在顯示區(qū)域允許的情況下，可以繼續(xù)按上述結(jié)構(gòu)增加。

繪制結(jié)果如下所示：