Как отобразить две разные легенды в hconcat chart с помощью altair - PullRequest
Новая
       28

Как отобразить две разные легенды в hconcat chart с помощью altair

0 голосов
/ 21 февраля 2020

Мне нужно отобразить две отдельные диаграммы вместе, включая их легенды в Jupyterlab, и единственный способ, которым мне удалось это сделать, - использовать hconcat. Я дошел до этого:

enter image description here

Однако даже с .resolve_legend(color='independent') я получаю записи из обеих диаграмм, отображаемых в обеих легендах вверху - что очень запутанно.

Результат должен выглядеть следующим образом:

enter image description here

Как я могу удалить ненужные записи легенды? Или, если кто-нибудь знает хорошую альтернативу, как составлять графики бок о бок в одной ячейке jupyterlab, я был бы рад пойти другим путем.

Мой код выглядит так: 

import altair as alt
import pandas as pd
from altair.expr import datum
df_test=pd.read_csv("test_df.csv")

chart_m1=alt.Chart(df_test).mark_bar().encode(
            x=alt.X('counts:Q', stack="normalize",axis=None),
            y=alt.Y('category:N',sort=['A','B','C'],title=None),
            color=alt.Color('grade:N',
                sort = alt.EncodingSortField( 'sort:Q', order = 'ascending' ),
                scale = alt.Scale(domain=['good <10', 'average 10-20', 'bad >20'], range=['#0cce6b', '#ffa400', '#ff4e42']),
                legend = alt.Legend(title="Metric1",orient='top')),
            order='sort:Q',
            tooltip=['category:N','grade:N','counts:Q']
            ).transform_filter(datum.metric=='metric1'
            ).properties(height=50,width=150)

chart_m2=alt.Chart(df_test).mark_bar().encode(
            x=alt.X('counts:Q', stack="normalize",axis=None),
            y=alt.Y('category:N',sort=['A','B','C'],title=None),
            color=alt.Color('grade:N',
                sort = alt.EncodingSortField( 'sort:Q', order = 'ascending' ),
                scale = alt.Scale(domain=['good <100', 'average 100-350', 'bad >350'], range=['#0cce6b', '#ffa400', '#ff4e42']),
                legend = alt.Legend(title="Metric2",orient='top')),
            order='sort:Q',
            tooltip=['category:N','grade:N','counts:Q']
            ).transform_filter(datum.metric=='metric2'
            ).properties(height=50,width=150)

alt.hconcat(chart_m1,chart_m2).resolve_legend(color='independent').configure_view(stroke=None)

test_df.csv Я использовал это: 

category,metric,sort,grade,counts
A,metric1,1,good <10,345
B,metric1,1,good <10,123
C,metric1,1,good <10,567
A,metric1,2,average 10-20,567
B,metric1,2,average 10-20,678
C,metric1,2,average 10-20,789
A,metric1,3,bad >20,900
B,metric1,3,bad >20,1011
C,metric1,3,bad >20,1122
A,metric2,1,good <100,1122
B,metric2,1,good <100,1011
C,metric2,1,good <100,900
A,metric2,2,average 100-350,789
B,metric2,2,average 100-350,678
C,metric2,2,average 100-350,567
A,metric2,3,bad >350,567
B,metric2,3,bad >350,345
C,metric2,3,bad >350,123

1 Ответ

3 голосов
/ 21 февраля 2020

Использование resolve_scale(color='independent')

alt.hconcat(
    chart_m1, chart_m2
).resolve_scale(
    color='independent'
).configure_view(
    stroke=None
)

enter image description here

Более подробная информация на https://altair-viz.github.io/user_guide/scale_resolve.html

...