Боке участок условный цвет фона


У меня есть фрейм данных pandas, содержащий такой логический столбец:

| A | B |   C   |
| 1 | 3 | True  |
| 2 | 4 | True  |
| 3 | 4 | False |
| 4 | 1 | False |
| 5 | 2 | True  |

И я хотел бы построить значения B на y и A на X с условным цветом фона на основе C Я имею в виду что-то вроде этого : смоделированный результат

Могу ли я сделать это с помощью аннотации коробки?

1   2   2017-04-07 16:29:43

1 ответ:

Да, это возможно, используя аргументы left и right BoxAnnotation:

import pandas as pd
from bokeh.plotting import figure, show, output_file, output_notebook
from bokeh.models import BoxAnnotation

output_notebook()

# dummy data 
df = pd.DataFrame({"A": [1, 2, 3, 4, 5, 6],
                   "B": [3, 4, 4, 1, 2, 3],
                   "C": [True, True, False, False, True, True]})
print(df)

>>>     A   B   C

    0   1   3   True
    1   2   4   True
    2   3   4   False
    3   4   1   False
    4   5   2   True
    5   6   3   True

Для упрощения я добавляю здесь еще одну строку, чтобы получить четные значения True для построения графика.

Теперь получим последовательные строки, содержащие True:

df["cons"] = (df["C"].diff(1) != 0).astype('int').cumsum() 
mask = df["cons"] % 2 == 1

cons_indices = df[mask].groupby("cons").apply(lambda x: x.A.values)
print(cons_indices)

>>> cons
    1    [1, 2]
    3    [5, 6]
    dtype: object

И, наконец, построить его:

p = figure(title="Annotations")
p.line(df["A"], df["B"])

for cons_index in cons_indices:
    low_box = BoxAnnotation(left=cons_index.min(), right=cons_index.max(), fill_color="Blue")
    p.add_layout(low_box)

show(p)

Решение пока не имеет дела с единичными True (не последовательными True) значениями. Однако вы не указали соответствующее поведение для этого сценария.

Введите описание изображения здесь