python - 具有渐变颜色的 Bokeh 散点图

Question

我正在尝试使用 Bokeh 绘制散点图。例如:

from bokeh.plotting import figure, show, output_notebook

TOOLS='pan,wheel_zoom,box_zoom,reset'
p = figure(tools=TOOLS)

p.scatter(x=somedata.x, y=somedata.y)

理想情况下，我希望在数据接近其最大/最小值 y 时以更强的强度着色。例如从红色到蓝色(-1 到 1)，就像在 heatmap 中一样(参数 vmax 和 vmin)。

有什么简单的方法吗？

Answer 1

Bokeh 具有将值映射到颜色的内置功能，然后将它们应用于绘图字形。

您也可以为每个点创建一个颜色列表，如果您不想使用此功能，则将其传入。

看下面一个简单的例子:

import numpy as np
from bokeh.plotting import figure, show
from bokeh.models import ColumnDataSource, LinearColorMapper


TOOLS='pan,wheel_zoom,box_zoom,reset'
p = figure(tools=TOOLS)

x = np.linspace(-10,10,200)
y = -x**2

data_source = ColumnDataSource({'x':x,'y':y})

color_mapper = LinearColorMapper(palette='Magma256', low=min(y), high=max(y))

# specify that we want to map the colors to the y values, 
# this could be replaced with a list of colors
p.scatter(x,y,color={'field': 'y', 'transform': color_mapper})

show(p)