python中matplotlib條件背景顏色的實現

這篇文章主要給大家介紹了關於python中matplotlib條件背景顏色的相關資料，文中通過示例代碼介紹的非常詳細，對大家學習或者使用python具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧

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

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', 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['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',linewidth=5,alpha=0.03)
for x in neg_1:
  ax.axvline(df.index[x], color='r',linewidth=5,alpha=0.03)

總結

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