如何使用Bokeh和Pandas绘制带有分类数据的散点图？

import pandas as pd
from bokeh.plotting import figure, show
from bokeh.io import output_notebook
from bokeh.models import CategoricalColorMapper

data = {'x': [1, 2, 3, 4, 5],
        'y': [6, 7, 8, 9, 10],
        'category': ['A', 'B', 'A', 'B', 'A']}
df = pd.DataFrame(data)

output_notebook()  # 在Jupyter Notebook中显示图表

# 创建一个Figure对象
p = figure()

# 创建一个分类颜色映射器
color_mapper = CategoricalColorMapper(factors=df['category'].unique(), palette=['red', 'blue'])

# 绘制散点图
p.circle(x='x', y='y', source=df, color={'field': 'category', 'transform': color_mapper}, legend_field='category')

# 设置图表属性
p.title.text = 'Scatter Plot with Categorical Data'
p.xaxis.axis_label = 'X'
p.yaxis.axis_label = 'Y'

# 显示图表
show(p)