Python bokeh: обновлять цвета точечной диаграммы при обратном вызове

Я только недавно начал использовать боке. У меня есть диаграмма рассеяния, на которой я хотел бы раскрасить каждый маркер в соответствии с определенным третьим свойством (скажем, количеством, в то время как ось x - это дата, а ось y - заданное значение в этот момент времени).

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

min_q = df.quantity.min()
max_q = df.quantity.max()
mapper = linear_cmap(field_name='quantity', palette=palettes.Spectral6, low=min_q, high=max_q)
source = ColumnDataSource(data=get_data(df))

p = figure(x_axis_type = "datetime")
p.scatter(x = "date_column", y = "value", marker = "triangle", fill_color=mapper, line_color=None, source=source)
color_bar = ColorBar(color_mapper=mapper['transform'], width=8,  location=(0,0))
p.add_layout(color_bar, 'right')

Кажется, это работает, как ожидалось. Ниже показан сюжет, который я получаю при запуске сервера боке.

Затем у меня есть функция обратного вызова Обновить(), запускаемая при изменении значения в некотором виджете (выбор или средство выбора времени).

def update():
    # get new df (according to new date/select)
    df = get_df()
    # update min/max for colormap
    min_q = df.quantity.min()
    max_q = df.quantity.max()
    # I think I should not create a new mapper but doing so I get closer
    mapper = linear_cmap(field_name='quantity', palette=palettes.Spectral6 ,low=min_q, high=max_q)
    color_bar.color_mapper=mapper['transform'] 
    source.data = get_data(df)
    # etc

Это самое близкое, что я мог найти. Цветовая карта обновляется с новыми значениями, но кажется, что цвета маркера по-прежнему соответствуют исходному шаблону. См. Рисунок ниже (учитывая это количество, я ожидал бы зеленого цвета, но он синий, поскольку он все еще отображается как <4000, как на карте первого графика перед обратным вызовом).

Должен ли я просто добавить во фрейм данных столбец «цвет»? Я считаю, что есть более простой / удобный способ сделать это.

РЕДАКТИРОВАТЬ: Вот минимальный рабочий пример с использованием ответа bigreddot:

from bokeh.io import curdoc
from bokeh.layouts import column
from bokeh.plotting import figure
from bokeh.models import Button, ColumnDataSource, ColorBar, HoverTool
from bokeh.palettes import Spectral6
from bokeh.transform import linear_cmap
import numpy as np

x = [1,2,3,4,5,7,8,9,10]
y = [1,2,3,4,5,7,8,9,10]
z = [1,2,3,4,5,7,8,9,10]



source = ColumnDataSource(dict(x=x, y=y, z=z))

#Use the field name of the column source
mapper = linear_cmap(field_name='z', palette=Spectral6 ,low=min(y) ,high=max(y))

p = figure(plot_width=300, plot_height=300, title = "Linear Color Map Based on Y")
p.circle(x='x', y='y', line_color=mapper,color=mapper, fill_alpha=1, size=12, source=source)

color_bar = ColorBar(color_mapper=mapper['transform'], width=8,  location=(0,0))
p.add_tools(HoverTool(tooltips = "@z", show_arrow=False, point_policy='follow_mouse'))
p.add_layout(color_bar, 'right')

b = Button()

def update():
    new_z = np.exp2(z)
    mapper = linear_cmap(field_name='z', palette=Spectral6 ,low=min(new_z), high=max(new_z))
    color_bar.color_mapper=mapper['transform'] 
    source.data = dict(x=x, y=y, z=new_z)

b.on_click(update)

curdoc().add_root(column(b, p))

После обновления круги будут окрашены в соответствии с исходной шкалой: все, что больше 10, будет красным. Вместо этого я бы ожидал, что все будет синим до последних 3 кружков на вершинах, которые должны быть окрашены в зеленый, желтый и красный цвета соответственно.