如何根據圖表中沒有的變數更改折線圖的背景顏色？例如，如果我有以下資料幀：

import numpy as np
import pandas as pd

dates = pd.date_range('20000101',periods=800)
df = pd.DataFrame(index=dates)
df['A'] = np.cumsum(np.random.randn(800)) 
df['B'] = np.random.randint(-1,2,size=800)

如果我做df.A的折線圖，如何根據該時間點'B'列的值更改背景顏色？

例如，如果在該日期B = 1，則該日期的背景為綠色。

如果B = 0，則該日期的背景應為黃色。

如果B = -1那麼背景那個日期應該是紅色的。

新增我最初考慮使用axvline的解決方法，但@jakevdp回答正是看起來因為不需要for迴圈：首先需要新增一個'i'列作為計數器，然後整個程式碼看起來像：

dates = pd.date_range('20000101',size=800)
df['i'] = range(1,801)

# getting the row where those values are true wit the 'i' value
zeros = df[df['B']== 0]['i'] 
pos_1 = df[df['B']==1]['i']
neg_1 = df[df['B']==-1]['i']

ax = df.A.plot()

for x in zeros:
 ax.axvline(df.index[x],color='y',linewidth=5,alpha=0.03)
for x in pos_1:
  ax.axvline(df.index[x],color='g',alpha=0.03)
for x in neg_1:
  ax.axvline(df.index[x],color='r',alpha=0.03)

總結

以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，謝謝大家對我們的支援。