Python plotlyを用いてガントチャートを作成するときにy軸の並び方を変更したい

Question

前提・実現したいこと

Pythonのplotlyの機能を使ってガントチャートが作りたいと思っています。
作業者と機械の稼働状況を確認したいのですが、作業者3人を上側に機械2台を下側にしたいです。

発生している問題・エラーメッセージ

y軸のタスクが、Worker1,Machine1_process,Worker2,Machine2_processと交互になっています。

該当のソースコード

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

def plot_guntt_chart(data):
    df = pd.DataFrame(columns = ["Task","Start","Finish","Resource","Job"])
    for i in data:
        df = df.append(data[i],ignore_index = True)
    fig = px.timeline(df, x_start="Start", x_end="Finish", y="Task",color = "Resource",text = "Job",title = "guntt chart")
    fig.update_yaxes(autorange="reversed")
    fig.show()

sample_data = {
1:{"Task":"Worker1", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"Machine1","Job":"JOB1"},
2:{"Task":"Worker2", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"Machine2","Job":"JOB2"},
3:{"Task":"Worker3", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"Machine3","Job":"JOB3"},
4:{"Task":"Worker3", "Start": datetime.datetime(2021,11,1,9), "Finish":datetime.datetime(2021,11,1,10),"Resource":"Machine3","Job":"JOB3"},
5:{"Task":"Machine1_process", "Start": datetime.datetime(2021,11,1,10), "Finish":datetime.datetime(2021,11,1,11),"Resource":"Machine1","Job":"JOB1"},
6:{"Task":"Worker3", "Start": datetime.datetime(2021,11,1,11), "Finish":datetime.datetime(2021,11,1,12),"Resource":"Machine3","Job":"JOB3"},
7:{"Task":"Machine2_process", "Start": datetime.datetime(2021,11,1,10), "Finish":datetime.datetime(2021,11,1,11),"Resource":"Machine2","Job":"JOB2"}
}
plot_guntt_chart(sample_data)

利用環境

python:3.7.11
plotly:5.1.0

Answer 1

update_yaxis(categoryorder=)で、直接順番を指定するのがいいのではないでしょうか。

def plot_guntt_chart(data):
    y_order = ['Worker1', 'Worker2', 'Worker3', 'Machine1_process', 'Machine2_process']
    df = pd.DataFrame(columns = ["Task","Start","Finish","Resource","Job"])
    for i in data:
        df = df.append(data[i],ignore_index = True)
    fig = px.timeline(df, x_start="Start", x_end="Finish", y="Task", color="Resource", text="Job", title="guntt chart")
    fig.update_yaxes(categoryorder='array', categoryarray=y_order[::-1])
    fig.show()

[::-1]はこれをつけないと下から上に整列されるためつけています。

Comments

[TakumiDeguchi]

回答ありがとうございます。
私の意図したとおりのガントチャートを作成することができました。

Answer 2

色が変わってよければ、以下のいずれかで可能です。

fig = px.timeline(df, x_start="Start", x_end="Finish", y="Task",color = "Task", text = "Job",title = "guntt chart")

fig = px.timeline(df, x_start="Start", x_end="Finish", y="Task" ,text = "Job",title = "guntt chart")

Comments

[TakumiDeguchi]

回答ありがとうございます。
こちらの方法でも実現できるか試してみます。
ご協力ありがとうございます。