從特定日期繪制不同顏色的折線圖

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()