我正在尝试使用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
)。
有一种简单的方法吗?
答案 0 :(得分:5)
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)