基于每个单元格不同范围的Bokeh DataTable颜色单元格

from bokeh.io import output_file, show
from bokeh.models import ColumnDataSource, DataTable, NumberFormatter, StringFormatter, HTMLTemplateFormatter
from bokeh.layouts import layout
from bokeh.palettes import RdYlGn10
from bokeh.plotting import curdoc

# 创建示例数据
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 30, 35, 40],
        'score': [80, 90, 70, 85]}

source = ColumnDataSource(data)

# 创建自定义的回调函数来设置单元格颜色
def color_mapper(value):
    if value < 70:
        return RdYlGn10[0]  # 红色
    elif value < 80:
        return RdYlGn10[4]  # 黄色
    else:
        return RdYlGn10[9]  # 绿色

# 创建格式化器和模板格式化器
age_formatter = NumberFormatter(format='0')
score_formatter = StringFormatter(font_style='bold')

score_template = """
<div style="background:<%= color_mapper(value) %>; color: white;"> <%= value %> </div>
"""

# 创建DataTable并设置列和格式化器
columns = [
    TableColumn(field='name', title='Name'),
    TableColumn(field='age', title='Age', formatter=age_formatter),
    TableColumn(field='score', title='Score', formatter=HTMLTemplateFormatter(template=score_template))
]

data_table = DataTable(source=source, columns=columns, index_position=-1)

# 将DataTable添加到文档中
curdoc().add_root(layout([[data_table]]))