python - 如何使用 pandas 数据框向绘图甘特图添加垂直线？

Question

我有一个数据集，其中包含列 Task(string 类型)、Period_Min 和 Period_Max。我想使用 plotly 绘制经典的甘特图，并为特殊的日子 2017-12-31 添加一条垂直线。

我已经按照plotly的官方doc尝试了基本的方法。 。

这是我的代码:

fig=ff.create_gantt(data_user1,colors['#333F44','#93e4c1'],index_col='Diff_temps')
layout = {
'shapes': [
    # Line Vertical
    {
        'type': 'line',
        'x0': '2017-12-31',
        'y0': data_user1.Task.values[0],
        'x1': '2017-12-31',
        'y1': data_user1.Task.tail(1).values[0],
        'line': {
            'color': 'rgb(55, 128, 191)',
            'width': 3,
        }
    }
]
}

iplot(fig, layout)

不幸的是，添加这个特定的布局不会改变我的图表中的任何内容。有什么帮助吗？

Answer 1

此Plotly community post中提供的提示在这里可能会有所帮助。

尝试一下:

fig = ff.create_gantt(data_user1, colors['#333F44','#93e4c1'], index_col='Diff_temps')

shapes = [
    # Line Vertical
    {
        'type': 'line',
        'x0': '2017-12-31',
        'y0': data_user1.Task.values[0],
        'x1': '2017-12-31',
        'y1': data_user1.Task.tail(1).values[0],
        'line': {
            'color': 'rgb(55, 128, 191)',
            'width': 3,
        }
    }
]
fig['layout']['shapes'] += shapes

iplot(fig, layout)