如何使用 matplotlib 或 plotly 放大 Python 中的图形？

Question

我想放大图表的红色部分。 我有以下代码片段。 我需要使用 matplotlib 或 plotly 在我的代码中进行哪些更改才能达到预期的结果。 如果可以，请首选 plotly 库。

# Visualize decision tree predictions

predictions = treePrediction
valid = df[x.shape[0]:]
valid["Predictions"] = predictions
plt.figure(figsize=(12, 7))
plt.title("Apple's Stock Price Prediction Model(Decision Tree Regressor Model)")
plt.xlabel("Days")
plt.ylabel("Close Price USD ($)")
plt.plot(df["Mean"])
plt.plot(valid[["Mean", "Predictions"]])
plt.legend(["Original", "Valid", "Predictions"])
plt.show()

Answer 1

参考已发布的节点书，我复制了有问题的图表。 由于 nodebook 使用的是区域设置 CSV 数据，因此我从yfinance检索了股票价格数据并将收盘价替换为df['Mean'] 。 Plotly，一个graph_object用于添加每个图。 我还在 select 中添加了一个按钮，用于缩放。 有关详细信息，请参阅此页面。 我们将添加 3 个月作为期间选择。

import plotly.graph_objects as go

fig = go.Figure()
fig.add_trace(go.Scatter(mode='lines', x=df.index, y=df['Mean'], line_color='blue', name='Original'))
fig.add_trace(go.Scatter(mode='lines', x=valid.index, y=valid['Mean'], line_color='orange', name='Valid'))
fig.add_trace(go.Scatter(mode='lines', x=valid.index, y=valid['Predictions'], line_color='green',name='Predictions'))

fig.update_layout(
    autosize=True,
    height=600,
    title="Apple's Stock Price Prediction Model(Decision Tree Regressor Model)",
    xaxis_title="Days",
    yaxis_title="Close Price USD ($)",
    template='plotly_white'
)

# Add range slider
fig.update_layout(
    xaxis=dict(
        rangeselector=dict(
            buttons=list([
                dict(count=1,
                     label="1m",
                     step="month",
                     stepmode="backward"),
                dict(count=3,
                     label="3m",
                     step="month",
                     stepmode="backward"),              
                dict(count=6,
                     label="6m",
                     step="month",
                     stepmode="backward"),
                dict(count=1,
                     label="YTD",
                     step="year",
                     stepmode="todate"),
                dict(count=1,
                     label="1y",
                     step="year",
                     stepmode="backward"),
                dict(step="all")
            ])
        ),
        rangeslider=dict(
            visible=True
        ),
        type="date"
    )
)

fig.show()

Select 最近一个月使用 Select 期间按钮。

图表底部的范围 slider 可用于 select 的任何范围。