Python＆amp; Matplotlib：多级树图图？

Question

我最近从https://www.kaggle.com/philippsp/exploratory-analysis-instacart

看到此图表

由R，

制作

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/

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()