python中的matplotlib條件背景顏色

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

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循環：首先需要添加一個'我'列作爲計數器，然後整個代碼如下所示：

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)

來源

2015-10-26 Gabriel

什麼樣的背景顏色？圖表？文本標籤？數據點本身的顏色？請舉個例子。 – MattDMo

圖表的背景顏色。正在考慮用垂直線做，但不知道它是否是最有效的方法。 – Gabriel

您可以使用繪圖命令之後pcolor()或pcolorfast()做到這一點。例如，使用數據，您上述定義：

ax = df['A'].plot() 
ax.pcolorfast(ax.get_xlim(), ax.get_ylim(), 
       df['B'].values[np.newaxis], 
       cmap='RdYlGn', alpha=0.3)

來源

2015-10-27 02:42:43 jakevdp

也許也想使用負Z順序。 – tacaswell

這正是我所需要的。謝謝@jakevdp – Gabriel

我想補充說，當你將不同的dfs（不同的時間軸）繪製到一個軸上時，這不能正常工作。 –

python中的matplotlib條件背景顏色

回答

相關問題