График как средний, так и средний на графике Альтаира

Question

Как можно изобразить среднее и медианное значение в Альтаире, отличающееся цветовой кодировкой?

Ниже моя первая попытка, но она не включает в себя легенду и не выглядит как самая элегантнаяпуть.

import altair as alt
from vega_datasets import data

source = data.cars()

mean = alt.Chart(source).mark_line(color='red', point=True).encode(
    x='Year',
    y='mean(Miles_per_Gallon)'
)

median = alt.Chart(source).mark_line().encode(
    x='Year',
    y='median(Miles_per_Gallon)'
)

mean+median

Answer 1

Лучше всего это сделать с помощью Aggregate Transform для вычисления агрегатов, за которым следует Fold Transform , чтобы два столбца можно было использовать в одной кодировке:

import altair as alt
from vega_datasets import data

source = data.cars()

alt.Chart(source).transform_aggregate(
    mean='mean(Miles_per_Gallon)',
    median='median(Miles_per_Gallon)',
    groupby=['Year']
).transform_fold(
    ['mean', 'median'],
    as_=['aggregate', 'value']
).mark_line().encode(
    x='Year',
    y='value:Q', 
    color='aggregate:N',
)