python - 在 altair 中向分层图表添加图例

Question

考虑以下示例：

    import altair as alt
    from vega_datasets import data

    df = data.seattle_weather()

    temp_max = alt.Chart(df).mark_line(color='blue').encode(
        x='yearmonth(date):T',
        y='max(temp_max)',
    )

    temp_min = alt.Chart(df).mark_line(color='red').encode(
        x='yearmonth(date):T',
        y='max(temp_min)',
    )

    temp_max + temp_min

在结果图表中，我想添加一个图例，显示蓝线显示最高温度，红线显示最低温度。实现这一目标的最简单方法是什么？

我看到（例如在这个问题的解决方案中：Labeling Layered Charts in Altair (Python)）如果在编码中设置颜色或大小左右，altair 只会添加一个图例，通常使用分类列，但这不是可能在这里，因为我正在绘制整个列，并且标签应该是列名（现在显示在 y 轴标签中）。

标签： pythonlegendaltair

Answer 1

我会做一个折叠变换，以便可以正确编码变量。

import altair as alt
from vega_datasets import data

df = data.seattle_weather()

alt.Chart(df).mark_line().transform_fold(
    fold=['temp_max', 'temp_min'], 
    as_=['variable', 'value']
).encode(
    x='yearmonth(date):T',
    y='max(value):Q',
    color='variable:N'
)