Условный цвет фона matplotlib в Python ⇐ Python

[Anonymous]

Как изменить цвет фона линейной диаграммы на основе переменной, которой нет в диаграмме?
Например, если у меня есть следующий фрейм данных:

Код: Выделить всё

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)

Подробнее здесь: https://stackoverflow.com/questions/333 ... -in-python