Как заполнить область разными цветами в зависимости от y в боке - PullRequest
Новая
       5

Как заполнить область разными цветами в зависимости от y в боке

1 голос
/ 23 апреля 2020

Цвет заливки должен быть зеленым при y> = 0 и красным при y <= 0. Вы можете сделать это в Matplotlib, используя переменную 'when' в fill_between. Боке имеет подобную функцию? </p> 

from bokeh.plotting import figure, output_file, show
import numpy as np

strike1 = 20 #Long Call
premium1 = 0.5
price = np.arange(15,25,0.01)
contracts = 1

def long_call(price, strike1, premium1, contracts):
    P = []
    for i in price:
        P.append((max(i - strike1, 0) - premium1) * (contracts * 100))
    return np.array(P)


# output to static HTML file
output_file("lines.html")

# create a new plot with a title and axis labels
p = figure(title="Option Payoff", x_axis_label='Underlying Price ($)', y_axis_label='Profit/Loss ($)')

# add a line renderer with legend and line thickness
p.line(x, y, line_width=2)
p.varea(x=x, y1=y, fill_alpha=1, fill_color='#3cb371')

# show the results
show(p)

1 Ответ

2 голосов
/ 23 апреля 2020

Глиф VArea является непрерывным, но его можно представить как отдельные части, просто сворачивая некоторые области в блок 0 областей, делая y1 и y2 одинаковыми с преобразованием. 

import math

from bokeh.models import ColumnDataSource, CustomJSTransform
from bokeh.plotting import figure, show
from bokeh.transform import transform

N = 100
ds = ColumnDataSource(dict(x=[i / 10 for i in range(N)],
                           y=[math.sin(i / 10) for i in range(N)]))
p = figure()
p.line('x', 'y', source=ds, line_width=3)
p.varea(x='x', y1=transform('y', CustomJSTransform(v_func="return xs.map(x => x > 0 ? x : 0)")),
        y2=0, source=ds, color='green', fill_alpha=0.5)
p.varea(x='x', y1=transform('y', CustomJSTransform(v_func="return xs.map(x => x < 0 ? x : 0)")),
        y2=0, source=ds, color='red', fill_alpha=0.5)

show(p)
...