在情节图中标记特殊日子

Posted 2021-04-30

我正在使用Plotly API从python中的数据集绘制折线图。例如，我有2018年的数据集。

现在我想查看天气周末天需求较高的工作日。我的目标是在星期六和星期日标记图表中的线条。 Plotly是否提供了这样做的方法？如果没有，有没有简单的方法来实现这一目标？

答案

您可以制作标记图，然后覆盖线图以获得所需的输出，其中必须使用基于条件的颜色列表传递标记图。

要获得清晰的图片，请参阅以下代码，

# make a dummy time data

x_values = pd.date_range(start=pd.Timestamp('2018-01-01'), end=pd.Timestamp('2019-01-01'), freq='1 D')

# making a random time series
y_values = np.random.randn(len(x_values))

# making color list
# red if the day is saturday or sunday else green

colors = ['red' if int(pd.Timestamp(d).weekday()) >= 5 else 'green' for d in x_values]

# make the plot

trace_0 = go.Scatter(
    x=x_values,
    y=y_values,
    mode='markers',
    marker=dict(
        color = colors
    )
)

trace_1 = go.Scatter(
    x=x_values,
    y=y_values,
    mode='lines',
    marker=dict(
        color = 'black'
    )
)

layout = dict(
    title='Time Series with Rangeslider',
    xaxis=dict(
        rangeselector=dict(
            buttons=list([
                dict(count=1,
                     label='1m',
                     step='month',
                     stepmode='backward'),
                dict(count=6,
                     label='6m',
                     step='month',
                     stepmode='backward'),
                dict(step='all')
            ])
        ),
        rangeslider=dict(
            visible = True
        ),
        type='date'
    )
)

fig = {
    'data': [trace_0, trace_1],
    'layout': layout,
}

py.iplot(fig, filename='Plot')

输出图

我还实现了一个范围滑块，如果不需要，可以从布局中编辑它。