Plotly 树状图
本教程将讨论在 Python 中使用 Plotly 的 treemap() 函数创建树状图。
在 Python 中使用 Plotly 的 treemap() 函数创建 Treemap 图
树状图将数据表示为图表上的嵌套矩形。我们可以使用 plotly.express 的 treemap() 函数在 Python 中创建树状图。要创建树形图，我们必须定义其父级的名称并将它们传递到 treemap() 函数中。
例如，让我们创建一个家庭的树形图，其中 Tony 有两个孩子，Emily 和 Draco，Emily 也有两个孩子，Andree 和 Jason，而 Draco 只有一个孩子，Peter。
我们将在列表中定义所有家庭成员的名称并将其传递给 names 参数，然后我们将定义每个成员的父级并将其传递给 parents 参数。
请参阅下面的代码。
import plotly.express as px
names = ["Tony", "Andree", "Jason", "Emily", "Petter", "Draco"]
parents = ["", "Emily", "Emily", "Tony", "Draco", "Tony"]
fig = px.treemap(names=names, parents=parents)
fig.show()
输出：
parents 列表中的第一个元素是空的，因为 Tony 没有父元素。我们可以使用 width 和 height 参数来设置绘图的宽度和高度。
使用颜色参数，我们可以使用默认颜色序列为每个矩形赋予不同的颜色。我们可以使用 title 参数给直方图一个标题。
我们可以使用 hover_name 参数更改悬停标题，并将其值设置为与输入数据大小相同的列表。我们可以使用 ids 参数给每个三角形一个不同的 id。
例如，让我们更改上面提到的属性。请参阅下面的代码。
import plotly.express as px
names = ["Tony", "Andree", "Jason", "Emily", "Petter", "Draco"]
parents = ["", "Emily", "Emily", "Tony", "Draco", "Tony"]
fig = px.treemap(
names=names,
parents=parents,
color=names,
title="Family Treemap",
hover_name=names,
ids=names,
)
fig.show()
输出：
我们可以使用 color_discrete_sequence 参数更改默认颜色序列。我们可以将颜色序列设置为 Plotly 支持的颜色序列，如 Dark24 或 Light24。
例如，让我们改变上面直方图的颜色顺序。请参阅下面的代码。
import plotly.express as px
names = ["Tony", "Andree", "Jason", "Emily", "Petter", "Draco"]
parents = ["", "Emily", "Emily", "Tony", "Draco", "Tony"]
fig = px.treemap(
names=names,
parents=parents,
color=names,
color_discrete_sequence=px.colors.qualitative.Light24,
)
fig.show()
输出：
除了使用颜色序列，我们还可以使用 color_discrete_map 参数为每个矩形赋予任何颜色。此参数用于覆盖默认颜色。
如果我们只改变几个矩形颜色，其余的将从颜色序列中分配。例如，让我们将前两个矩形颜色更改为黄色和黑色。
请参阅下面的代码。
import plotly.express as px
names = ["Tony", "Andree", "Jason", "Emily", "Petter", "Draco"]
parents = ["", "Emily", "Emily", "Tony", "Draco", "Tony"]
fig = px.treemap(
names=names,
parents=parents,
color=names,
color_discrete_sequence=px.colors.qualitative.Light24,
color_discrete_map={"Tony": "yellow", "Emily": "black"},
)
fig.show()
输出：
我们还可以使用数据框来创建树状图。例如，让我们阅读 2007 年的世界预期寿命数据，并创建它的树状图，其中以大陆名称作为父母，以国家名称作为孩子。
请参阅下面的代码。
import plotly.express as px
df = px.data.gapminder().query("year == 2007")
fig = px.treemap(
df,
path=[px.Constant("world"), "continent", "country"],
color="lifeExp",
hover_data=["country"],
color_continuous_scale="RdBu",
)
fig.show()
输出：
