python - Holoviews，为 HoloMap 的每一帧叠加不同的 VLine

Question

我正在尝试使用以下代码创建散点图以及 x 轴值平均值的垂直线，但我需要一些帮助来实现它。

import pandas as pd
import holoviews as hv
hv.extension('bokeh')

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

m1 = df['x'].mean()
line1 = hv.VLine(m1)
line1.opts(color='red', line_dash='dashed')    
ds = hv.Dataset(df, kdims=['group'], vdims=['x', 'y'])
scatter = ds.to(hv.Scatter, 'x', 'y')
scatter.opts(width=400, height=400, size=10,
             tools=['hover'], xlim=(0, 11), ylim=(0, 11))
layout = scatter * line1
hv.save(layout, 'means0.html')

它按照散点图的预期工作得很好，当更改“组”维度时，会更改绘制的点，但平均值的垂直线会针对整个数据帧计算一次，我也不知道如何实现该元素的交互性。这意味着每当我更改“组”维度来计算所选“组”的平均值并像对点一样绘制它时。

Answer 1

你就快到了，你只需要将每个 VLine(以及相应计算的子集的平均值)放入 HoloMap 中即可。下面通过字典显式初始化 HoloMap。

import pandas as pd
import holoviews as hv
hv.extension('bokeh')

df = pd.DataFrame({'group': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
                   'x': [3, 4, 5, 2, 2, 3, 5, 4, 7],
                   'y': [10, 8, 9, 3, 7, 6, 5, 4, 8]})
hm = hv.HoloMap({
    group: hv.Scatter(df.loc[df.group==group], 'x', 'y') * hv.VLine(df.loc[df.group==group, 'x'].mean())
    for group in df.group.unique()
})

hm.opts(
    opts.Scatter(width=400, height=400, size=10,
                 tools=['hover'], xlim=(0, 11), ylim=(0, 11)),
    opts.VLine(color='red', line_dash='dashed') 
)