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

Question

前提・実現したいこと

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

発生している問題

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

該当のソースコード

import pandas as pd
import datetime 
import plotly.express as px
import plotly.io as pio


y_order = ["Worker3","Worker2","Worker1"]
sample_data = {1:{"Task":"Worker1", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"MachineB","Job":"JOB1"},
              2:{"Task":"Worker2", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"MachineA","Job":"JOB2"},
               3:{"Task":"Worker3", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"MachineC","Job":"JOB3"},
              }
def plot_gantt_chart(data,y_order):
    df = pd.DataFrame(columns = ["Task","Start","Finish","Resource"])
    for i in data:
        df = pd.concat([df,pd.DataFrame.from_dict(data[i], orient='index').T])
    fig = px.timeline(df, x_start="Start", x_end="Finish", y="Task",color = "Resource",title = "guntt chart")
    fig.update_yaxes(autorange="reversed",categoryorder='array', categoryarray=y_order[::]) 
    fig.show()
plot_gantt_chart(sample_data,y_order)

補足情報

python= 3.9.7
plotly= 5.6.0
plotly_express = 0.4.1

Answer 1

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

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

Comments

[TakumiDeguchi]

ありがとうございます。順番通りに並べることができました。