绘制甘特图：删除重复的Y轴标签并堆叠并行任务

Question

我计划使用python库“Plotly”构建甘特图。具体而言：https://plotly.com/python/gantt/#group-一起完成任务。

然而，每个“作业”可以有多个任务，并且这些任务可以并行运行。从我观察到的情况来看，Plotly并没有将并行运行的任务堆叠在一起，这使得读取图表非常困难。下面是一个示例，“作业A”有两个并行运行的任务，但只有一个可见：

data = [dict(Task="Job A", Start='2009-01-01', Finish='2009-02-28'),
      dict(Task="Job A", Start='2009-01-01', Finish='2009-02-28'),
      dict(Task="Job B", Start='2009-03-05', Finish='2009-04-15'),
      dict(Task="Job C", Start='2009-02-20', Finish='2009-05-30')]

# Without group_tasks=True, There would be two separate "Job A" labels
fig = ff.create_gantt(data, group_tasks=True)
fig.show()

我想要的是“作业A”任务都是可见的，但要垂直堆叠，“作业A”位于其任务占用的垂直空间的中心。类似这样，但没有两个“工作A”标签：

如果有人对我的甘特图项目有任何库建议，我应该考虑，请随时分享！谢谢！

Answer 1

import plotly.express as px
import pandas as pd

data = [dict(Task="Job A", Start='2009-01-01', Finish='2009-02-28'),
      dict(Task="Job A", Start='2009-01-01', Finish='2009-02-28'),
      dict(Task="Job B", Start='2009-03-05', Finish='2009-04-15'),
      dict(Task="Job C", Start='2009-02-20', Finish='2009-05-30')]

df = pd.DataFrame(data)

df['JobNum'] = ""
df.loc[0,'JobNum'] = 1
for idx in range(1,df.shape[0]):
    if df.loc[idx-1,'Task'] == df.loc[idx,'Task']:
        df.loc[idx,'JobNum'] = df.loc[idx-1,'JobNum'] + 1
    else:
        df.loc[idx,'JobNum'] = 1

df['hoverName'] = df.apply(lambda x: x['Task'] + "|" + str(x['JobNum']), axis=1)

方法1：使用Facet行

fig = px.timeline(df
                  , x_start="Start"
                  , x_end="Finish"
                  , y="Task"
                  , hover_name= "Task"
                  , color_discrete_sequence=px.colors.qualitative.Prism
                  , opacity=.7
                  , template='plotly_white'
                  , color='Task'
                  , facet_row= 'JobNum'
                  , hover_data = ['Start','Finish']
                 )
fig.show()

Appraoch 1输出

近似值2：调整宽度和偏移。当有两个以上的并行任务时，这需要推广。

fig = px.timeline(df
                  , x_start="Start"
                  , x_end="Finish"
                  , y="Task"
                  , hover_name= "hoverName"
                  , color_discrete_sequence=px.colors.qualitative.Prism
                  , opacity=.7
                  , template='plotly_white'
                  , color='JobNum'
                  , hover_data = ['Start','Finish']
                 )

for obj in fig.data:
    Task, JobNum = obj.hovertext[0].split("|")
    if (int(JobNum) == 1):
        obj.width = 0.1
        obj.offset = 0.05
    elif (int(JobNum) == 2):
        obj.width = 0.1
        obj.offset = -0.05

fig.show()

方法2输出

Answer 2

一个起点是使用fig.add_shape添加一个相同的任务作为原始任务下方的矩形。

要做到这一点，我们需要每个矩形的y坐标，但方便的是，第一个条位于y=0，第二个条位于y=1，依此类推。因此，按顺序列出的唯一任务的索引也是y坐标（唯一任务是[作业A、作业B、作业C]，因此作业C栏将居中于y=3）。每个条的默认宽度为0.8，因此如果y0是条的起始y坐标，则y1应结束于y0-0.4。

请注意，带注释的形状没有任何悬停模板，每个条的颜色与当前编写的颜色相同。

import numpy as np
import pandas as pd
import plotly.express as px
import plotly.graph_objects as go

## added additional duplicate Task to demonstrate generalizability
data = [dict(Task="Job A", Start='2009-01-01', Finish='2009-02-28'),
      dict(Task="Job A", Start='2009-01-01', Finish='2009-02-28'),
      dict(Task="Job B", Start='2009-03-05', Finish='2009-04-15'),
      dict(Task="Job C", Start='2009-02-20', Finish='2009-05-30'),
      dict(Task="Job C", Start='2009-02-20', Finish='2009-05-30')]

df = pd.DataFrame(data)

# Without group_tasks=True, There would be two separate "Job A" labels
# fig = ff.create_gantt(data, group_tasks=True)

## plot the non-duplicate rows
fig = px.timeline(df.loc[~df['Task'].duplicated()], x_start="Start", x_end="Finish", y="Task")

## plot the duplicate rows using rectangular shapes
for row in df.loc[df['Task'].duplicated()].itertuples():
      y_val = np.where(df.Task.unique()==row[1])[0][0]
      # print(f"found {row[1]} at index {y_val}")
      fig.add_shape(type="rect",
            xref="x", yref="y",
            x0=row[2], x1=row[3], 
            y0=y_val, y1=y_val-0.4,
            line_width=0,
            fillcolor="salmon",
)
fig.show()