Python + Matplotlib：多层级树图绘制？

Question

我最近看到了这个来自https://www.kaggle.com/philippsp/exploratory-analysis-instacart的矩形树图表格（两级层次结构，有颜色，方块化的矩形树图）。

它是用 R ggplot2::treemap 制作的，由此产生:

treemap(tmp,index=c("department","aisle"),vSize="n",title="",
        palette="Set3",border.col="#FFFFFF")

我想知道如何在Python中制作这个图表？

---

我搜索了一下，但没有找到任何多级树状图的例子。

• https://gist.github.com/gVallverdu/0b446d0061a785c808dbe79262a37eea
• https://python-graph-gallery.com/200-basic-treemap-with-python/

- cqcn1991

---

你是否曾经在Python中找到过一个多级树状图库？ - Mark K Cowan

Answer 1

你可以使用 Plotly。在这里，你可以找到几个示例。

https://plotly.com/python/treemaps/

这是一个非常简单的多层结构示例。

import plotly.express as px
import pandas as pd
from collections import defaultdict

data = defaultdict()

data['level_1'] = ['A', 'A', 'A', 'B', 'B', 'B']
data['level_2'] = ['X', 'X', 'Y', 'Z', 'Z', 'X']
data['level_3'] = ['1', '2', '2', '1', '1', '2']

data =  pd.DataFrame.from_dict(data)
fig = px.treemap(data, path=['level_1', 'level_2', 'level_3'])
fig.show()

- milihoosh

Answer 2

包matplotlib-extra提供了一个支持多级树状图绘制的treemap函数。对于G20数据集，treemap可以生成类似的树状图，例如：

import matplotlib.pyplot as plt
import mpl_extra.treemap as tr

fig, ax = plt.subplots(figsize=(7,7), dpi=100, subplot_kw=dict(aspect=1.156))

trc = tr.treemap(ax, df, area='gdp_mil_usd', fill='hdi', labels='country',
           levels=['region', 'country'],
           textprops={'c':'w', 'wrap':True,
                      'place':'top left', 'max_fontsize':20},
           rectprops={'ec':'w'},
           subgroup_rectprops={'region':{'ec':'grey', 'lw':2, 'fill':False,
                                         'zorder':5}},
           subgroup_textprops={'region':{'c':'k', 'alpha':0.5, 'fontstyle':'italic'}},
           )

ax.axis('off')

cb = fig.colorbar(trc.mappable, ax=ax, shrink=0.5)

cb.ax.set_title('hdi')
cb.outline.set_edgecolor('w')

plt.show()

获取到的树状图如下：

如需更多信息，您可以查看项目，其中有一些示例。源代码中包含API文档字符串。

- Z-Y.L