python - 在 Altair 中合并两个图例

Question

我在 Altair 中有一个散点图，其中我使用形状和颜色来表示一列。我想要一个包含两条信息的图例，但我得到了两个图例，一个用于形状，另一个用于颜色。

代码如下。请参阅this notebook获取可重现的示例(您需要输入 Google 凭据才能加载数据)。

import altair as alt
alt.themes.enable('fivethirtyeight')
selection = alt.selection_multi(fields=['Domain'], bind='legend')

chart = alt.Chart(df, width=1100, height=600, 
          title="Parameter count of ML systems through time")\
.mark_point(size=120, filled=False).encode(
  x=alt.X('Publication date:T'),
  y=alt.Y('Parameters:Q',
          scale=alt.Scale(type='log', domain=(1, 3e13)), 
          axis=alt.Axis(format=".1e")),
  color=alt.Color('Domain',  
                  sort=['Vision', 'Language', 'Games', 'Other'],
                  legend=alt.Legend(
                      values = ['Vision', 'Language', 'Games', 'Other'],),),
  shape = alt.Shape('Domain'),#, legend=None),
  tooltip=['System', 
           'Reference', 
           'Publication date', 
           alt.Tooltip('Parameters', format=".1e"), 
           'Domain'],
  opacity=alt.condition(selection, alt.value(1), alt.value(0.2))
)

regression = chart.transform_regression(
    on="Publication date", 
    regression="Parameters",  
    method = 'exp',
    groupby=["Domain"],
).mark_line(point=False, strokeDash=[10,5], clip=True)

alt.layer(chart.add_selection(selection), regression).configure_axis(
    labelFontSize=20,titleFontSize=30).configure_legend(
    titleFontSize=20,
    labelFontSize =18,
    gradientLength=400,
    gradientThickness=30,
    symbolSize = 130,
)

如何将两个图例合并为一个图例？

Answer 1

您可以在折线图中将形状和颜色的图例设置为None，然后根据问题的评论使用resolve_scale:

import altair as alt
from vega_datasets import data


df = data.cars()
selection = alt.selection_multi(fields=['Origin'], bind='legend')

chart = alt.Chart(df).mark_point(filled=False).encode(
  x=alt.X('Acceleration'),
  y=alt.Y('Horsepower',scale=alt.Scale(type='log'), axis=alt.Axis(format=".1e")),
  color='Origin',
  shape='Origin',
  opacity=alt.condition(selection, alt.value(1), alt.value(0.2))
)

regression = chart.transform_regression(
    on="Acceleration", regression="Horsepower", groupby=["Origin"]
).mark_line(
).encode(color=alt.Color('Origin', legend=None), shape=alt.Shape('Origin', legend=None))

(alt.layer(chart, regression)
 .resolve_scale(shape='independent', color='independent')
 .add_selection(selection))