在Altair中连接两个图表时控制图例的颜色和顺序

时间:2019-04-21 13:47:21

标签: altair

我很难在图例中获得正确的颜色顺序。

我正在尝试达到:

cash: blue
fixed_income: yellow
equity: red

我正在使用以下数据框'dfl'绘制三个图表的两行:

  trade_date account owner account_type asset  value  sort_asset
0 2002-01-02  p2_inv    p2          inv  cash    0.0           0
1 2002-01-03  p2_inv    p2          inv  cash    0.0           0
2 2002-01-04  p2_inv    p2          inv  cash    0.0           0

dfl.shape(76824,7)

我有以下代码:

df_p1 = dfl[dfl['owner'] == 'p1']
df_p2 = dfl[dfl['owner'] == 'p2']

base_p1 = alt.Chart(df_p1).mark_area().encode(
    x=alt.X('trade_date:T', title=""), 
    y=alt.Y('sum(value):Q', stack='normalize', title="Asset Allocation"), 
    color=alt.Color('asset:N', sort=['cash', 'fixed_income', 'equity']),
    order=alt.Order('sort_asset:N', sort='ascending')
).properties(
    width=120,
    height=160
).facet(
    column=alt.Column('account:N'), 
)
base_p2 = alt.Chart(df_p2).mark_area().encode(
    x=alt.X('trade_date:T', title=""), 
    y=alt.Y('sum(value):Q', stack='normalize', title="Asset Allocation"), 
    color=alt.Color('asset:N', sort=['cash', 'fixed_income', 'equity']),
    order=alt.Order('sort_asset:N', sort='ascending')
).properties(
    width=120,
    height=160
).facet(
    column=alt.Column('account:N'),
)

base_p1 & base_p2

当我自己运行base_p1或base_p2时,我得到了正确的图例。但是当我使用&将他们加入一起时,我的图例和颜色将更改为:

cash: blue
equity: yellow
fixed income: red

我还注意到,我在DataFrame的sort_asset列中添加了可用于正确排序资产的信息,并且正在使用它来确保堆栈已正确排序。

我确定自己缺少一些简单的东西,因为我是altair的新手。我可以在上面的代码中进行哪些更改以呈现正确的颜色和顺序?

1 个答案:

答案 0 :(得分:1)

这是Altair版本2中的一个已知错误:排序字段未保留在复合图中。一些详细信息(包括解决方法的想法)在这里:https://github.com/altair-viz/altair/issues/820

Altair 3.0修复了该错误,应在下周的某个时间发布。