从特定日期绘制不同颜色的折线图

Question

绘图时我正在对数据集进行预测，我想在 python 中以不同颜色绘制来自特定日期的预测值。

我想要这样

提前致谢

Answer 1

您可以创建一个布尔数组来告诉要绘制哪个部分。

这是一个假设您使用熊猫日期的示例。 当<=和>=与作为 x 轴一部分的日期一起使用时，该行将继续。 如果没有，线路将有一个小的中断。

from matplotlib import pyplot as plt
import pandas as pd
import numpy as np

t = pd.date_range('2015', '2021', freq='M')
sales = np.random.normal(100, 500, len(t) ).cumsum() + 10000

past = t <= '2020-02-29'
future = t >= '2020-02-29'
plt.plot(t[past], sales[past], color='dodgerblue')
plt.plot(t[future], sales[future], color='crimson')
plt.fill_between(t[future], sales[future] - 1000, sales[future] + 1000, color='crimson', alpha=0.3)
plt.ylabel('sales')

plt.show()