Фоновая форма графика на основе переменной

Question

У меня есть пандас Dataframe, в котором есть несколько столбцов с данными и один столбец, который кодирует состояние интересующего процесса (непоследовательные целые числа).

Вместо того, чтобы отображать столбец состояния в виде линии, я бынравится использовать его, чтобы добавить тень к фону графика, например, следующим образом:

Пример кадра данных:

df = pd.DataFrame(
{
    "y": [x * x / 100 for x in range(10)],
    "state": [0 if x < 5 else 1 for x in range(10)],
})
    y   state
0   0.00    0
1   0.01    0
2   0.04    0
3   0.09    0
4   0.16    0
5   0.25    1
6   0.36    1
7   0.49    1
8   0.64    1
9   0.81    1

Желаемый график (обратите внимание, что состояние включается в виде строки для получениячерез точку, в окончательной картинке я, конечно, ее опущу):

Answer 1

Вы можете найти блоки, где state - постоянная величина, а затем использовать axvspan, чтобы заполнить эти блоки разными цветами:

fig, ax = plt.subplots(1)
ax.set_ylim(0,1)

df[['y']].plot(ax=ax)

x = df.loc[df['state'] != df['state'].shift(1), 'state'].reset_index()
x['next_index'] = x['index'].shift(-1).fillna(df.index.max())

for i in x.index:
    c = 'blue' if (x.at[i, 'state']==1) else 'red'
    xa = x.at[i, 'index']
    xb = x.at[i, 'next_index']
    ax.axvspan(xa, xb, alpha=0.15, color=c)

Выход: