Plotly 树状图

Ammar Ali 2022年5月14日
Plotly 树状图

本教程将讨论在 Python 中使用 Plotly 的 treemap() 函数创建树状图。

在 Python 中使用 Plotly 的 treemap() 函数创建 Treemap 图

树状图将数据表示为图表上的嵌套矩形。我们可以使用 plotly.expresstreemap() 函数在 Python 中创建树状图。要创建树形图,我们必须定义其父级的名称并将它们传递到 treemap() 函数中。

例如,让我们创建一个家庭的树形图,其中 Tony 有两个孩子,EmilyDracoEmily 也有两个孩子,AndreeJason,而 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 支持的颜色序列,如 Dark24Light24

例如,让我们改变上面直方图的颜色顺序。请参阅下面的代码。

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

输出:

数据框树状图

Author: Ammar Ali
Ammar Ali avatar Ammar Ali avatar

Hello! I am Ammar Ali, a programmer here to learn from experience, people, and docs, and create interesting and useful programming content. I mostly create content about Python, Matlab, and Microcontrollers like Arduino and PIC.

LinkedIn Facebook

相关文章 - Plotly Plot