主页 / coding / 问题 / 78824675
Accepted
Gerard L
Asked: 2024-08-02 17:16:35 +0800 CST

Plotly 时间线为每个资源绘制多个任务

  • 772

我有一个数据框形式的数据集，其结构如下

工作 任务 机器 开始 结尾 颜色
A 做 X 1 0 5 蓝色的
A 是吗 2 6 14 蓝色的
乙 做问 3 0 8 绿色的
乙 做 Z 3 9 12 绿色的
乙 做 X 1 14 17 绿色的
C 是吗 1 6 9 红色的

我想使用 Plotly（例如 px.timeline）在交互式时间线上绘制这些内容，类似于下面的图片，也称为甘特图。对于每台机器（y 轴），我想要一个彩色条形图来表示分配给该机器的任务，类似于其持续时间。对于我的示例数据，这意味着对于机器 1，有三个彩色条形图（0-5、6-9、14-17）。任务应根据定义的作业颜色着色，如颜色列中所述。理想情况下，将鼠标悬停在条形图上时，它会显示Job - Task, Start - end。

示例图

我使用以下内容；

Plotly 5.18.0
Python 3.9
Pandas 2.1.2

有没有办法做到这一点？

我尝试了 Plotly，但只设法在 y 轴上获取作业（https://plotly.com/python/gantt/）。我使用了https://plotly.com/python/figure-factories/，因为 px.timeline 似乎不接受开始和结束时间的整数值。

plotly

2 个回答

  1. Best Answer

    在 中px.timeline，无法指定颜色，因此我们只需使用水平条形图即可。要绘制具有指定颜色的水平条形图，请在循环过程中指定堆叠条形图，内容由 Job 有条件地从数据框中提取。指定基数是因为始终需要设置起始位置。文本显示只能选择内部和外部，因此我们使用字符串注释将其添加到条形图的中心

    import plotly.graph_objects as go

fig = go.Figure()

for j in df['Job'].unique():
    dfj = df.query('Job == @j')
    for row in dfj.itertuples():
        print(row)
        before_base = row.Start
        fig.add_trace(go.Bar(
            base=[row.Start],
            x0=[row.Start],
            x=[row.End- before_base],
            y=[row.Job],
            y0=[row.Job],
            hovertemplate='Start: '+str(row.Start)+'<br>'+
            'End: %{x}<br>'+row.Task+'<extra></extra>',
            orientation='h',
            #text=row.Task,
            marker_color=row.color,
            width=0.9,
            showlegend=False,
        ))
        fig.add_annotation(
            x=(row.Start+row.End)/2,
            y=row.Job,
            text=row.Task,
            font=dict(color='white'),
            showarrow=False,
        )
        fig.update_layout(barmode='stack')
    
fig.update_yaxes(autorange="reversed")
fig.update_layout(height=400)
fig.show()

    在此处输入图片描述

    • 0

  2. 水平条形图应该可以工作，您可以df使用来分析属于特定机器的每个部分df.groupby('Machine')，然后绘制go.Bar与该组内每一行相对应的轨迹（使用该行中的所有信息来确定每个条形的颜色、位置和持续时间）。

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

df = pd.DataFrame({
    'Job': ['A','A','B','B','B','C'],
    'Task': ['Do X','Do Y','Do Q','Do Z','Do X','Do Y'],
    'Machine': ['1','2','3','3','1','1'],
    'Start':[0,6,0,9,14,6],
    'End':[5,14,8,12,17,9],
    'color':['blue','blue','green','green','green','red']
})

df['Duration'] = df['End'] - df['Start']

fig = go.Figure()
for group_name, df_group in df.groupby('Machine'):
    for row_index, row in df_group.iterrows():
        start, duration, task, color = row['Start'], row['Duration'], row['Task'], row['color']
        
        # hovertemplate: Job - Task, Start - end
        customdata = [[row['Job'], row['Task'], row['Start'], row['End']]]  # Custom data for each trace
        fig.add_trace(
            go.Bar(
                x=[duration],
                base=[start],
                marker=dict(color=color),
                y=[f"M{group_name}"],
                orientation='h',
                showlegend=False,
                customdata=customdata,  
                hovertemplate="%{customdata[0]} - %{customdata[1]}, %{customdata[2]} - %{customdata[3]}<extra></extra>"
            )
        )
        fig.add_annotation(
            x=start+(duration/2),
            y=f"M{group_name}",
            text=task,
            font=dict(color='white'),
            showarrow=False,
        )

fig.update_layout(
    xaxis_title='time', yaxis_title='machine',
    barmode='stack', bargroupgap=0.0, bargap=0.0
)
fig.show()

    在此处输入图片描述

    • 0

相关问题