plotlyを用いて、凡例の順番を指定したガントチャートを作成したい

前提・実現したいこと

ガントチャートを作成するときに、図の右側の凡例の順番を指定したい。

発生している問題

color(Resouce)の凡例の順番がABC順にならない

該当のソースコード

Python
1import pandas as pd
2import datetime 
3import plotly.express as px
4import plotly.io as pio
5
6
7y_order = ["Worker3","Worker2","Worker1"]
8sample_data = {1:{"Task":"Worker1", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"MachineB","Job":"JOB1"},
9              2:{"Task":"Worker2", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"MachineA","Job":"JOB2"},
10               3:{"Task":"Worker3", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"MachineC","Job":"JOB3"},
11              }
12def plot_gantt_chart(data,y_order):
13    df = pd.DataFrame(columns = ["Task","Start","Finish","Resource"])
14    for i in data:
15        df = pd.concat([df,pd.DataFrame.from_dict(data[i], orient='index').T])
16    fig = px.timeline(df, x_start="Start", x_end="Finish", y="Task",color = "Resource",title = "guntt chart")
17    fig.update_yaxes(autorange="reversed",categoryorder='array', categoryarray=y_order[::]) 
18    fig.show()
19plot_gantt_chart(sample_data,y_order)

補足情報

python= 3.9.7
plotly= 5.6.0
plotly_express = 0.4.1

回答1件

ベストアンサー

trace オブジェクトを name(ここでは Resource 名)でソートします。

python
1def plot_gantt_chart(data,y_order):
2    df = pd.DataFrame(columns = ["Task","Start","Finish","Resource"])
3    for i in data:
4        df = pd.concat([df,pd.DataFrame.from_dict(data[i], orient='index').T])
5    fig = px.timeline(df, x_start="Start", x_end="Finish", y="Task",color = "Resource",title = "guntt chart")
6    fig.update_yaxes(autorange="reversed",categoryorder='array', categoryarray=y_order[::])
7    ##  追加
8    fig.data = sorted(fig.data, key=lambda x: x['name'])
9    fig['layout'].update(legend=dict(traceorder='normal'))
10    ##
11    fig.show()