python - Altair 置信区间图

Question

我希望使用 Altair 创建一个类似于附图的绘图，该图是使用 matplotlib 绘制的。基本上，一条绘制平均值的线和一个置信区间。

我的数据框看起来像这样。

dfr = pd.DataFrame(list(zip(mean, lower, upper)),
                       columns=["Mean", "Lower", "Upper"],
                       index=pd.RangeIndex(46, name="x"))

知道如何实现这一目标吗？我无法弄清楚 Altair 的语法。谢谢

Answer 1

您可以通过将线图和面积图分层来完成此操作。例如:

import altair as alt
import numpy as np
import pandas as pd

np.random.seed(9)
mean = np.random.randn(46).cumsum()
lower = mean - 2
upper = mean + 2

dfr = pd.DataFrame(list(zip(mean, lower, upper)),
                       columns=["Mean", "Lower", "Upper"],
                       index=pd.RangeIndex(46, name="x"))

line = alt.Chart(dfr.reset_index()).mark_line().encode(
    x='x',
    y='Mean'
)

band = line.mark_area(opacity=0.5).encode(
    x='x',
    y='Lower',
    y2='Upper'
)

band + line