python - 如何使用 Bokeh 直接显示 statsmodels Mosaics

Question

我正在寻找创建一些马赛克图来可视化列联表。Bokeh 本身并不支持马赛克图（据我所知），所以我使用的是statsmodels库（链接）。

问题是该库中的图没有扩展 Bokeh 的Figure界面，所以我无法让它们显示在网页中。理想情况下，我希望用户能够使用网页上的下拉框选择他们感兴趣的变量：

mosaicBokeh如何直接显示statsmodels的结果？

Answer 1

Bokeh 不直接支持马赛克图表，但是该mosaic函数可以返回让 Bokeh 自己渲染绘图所需的所有几何数据。此外，如果您通过ax=None：

_, rects_dict = mosaic(df, ..., ax=None)

那么 Matplotlib 图的生成将被抑制。下面是一个完整的示例，演示了如何使用rects_dictfrom 的返回值mosaic：

import pandas as pd
from statsmodels.graphics.mosaicplot import mosaic
from bokeh.plotting import figure, ColumnDataSource, show
from bokeh.transform import factor_cmap

df = pd.DataFrame({
    'size' : ['small', 'medium', 'medium', 'large', 'small', 'large', 'small', 'medium'],
    'length' : ['long', 'short', 'long', 'short', 'long', 'long', 'short', 'short']
})

_, rects_dict = mosaic(df, ['size', 'length'], gap=0, ax=None)

rects = rects_dict.values()
cats = rects_dict.keys()

source = ColumnDataSource(data=dict(
    x    = [r[0]+r[2]/2 for r in rects], # bokeh wants x center, not corner
    y    = [r[1]+r[3]/2 for r in rects], # bokeh wants y center, not corner
    w    = [r[2]        for r in rects],
    h    = [r[3]        for r in rects],
    size = [c[0]        for c in cats ],
    len  = [c[1]        for c in cats ],
))

fill_cmap = factor_cmap('size', palette="Pastel1_3", factors=['small', 'medium', 'large'])

p = figure(x_range=(0,1), y_range=(0,1), x_axis_location=None, y_axis_location=None,
           tools="", toolbar_location=None, tooltips="@size @len")

p.rect(x='x', y='y', width='w', height='h', line_color="white", source=source,
       fill_color=fill_cmap)

show(p)

这会导致下图带有显示类别的交互式悬停工具提示。您还可以直接以标准 Bokeh 方式添加颜色条或任何其他 Bokeh 交互功能：