Altair:如何根据线条的最大值，在刻面网格中设置不同的线条样式？

import altair as alt

# 创建一个示例数据集
data = alt.Data(values=[
    {"x": 1, "y": 5},
    {"x": 1, "y": 8},
    {"x": 2, "y": 3},
    {"x": 2, "y": 6},
    {"x": 3, "y": 4},
    {"x": 3, "y": 7}
])

# 使用 transform_aggregate() 计算每个刻面中线条的最大值
agg_data = data.transform_aggregate(
    max_y='max(y)',
    groupby=['x']
)

# 使用 mark_rule() 创建线条图，并设置不同的颜色
chart = alt.Chart(agg_data).mark_rule().encode(
    x='x:O',
    y='max_y:Q',
    color=alt.Color('max_y:Q', scale=alt.Scale(scheme='viridis'))
)

# 显示图表
chart.show()