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()
輸出: