Условный поиск Альтаир ⇐ Python

[Anonymous]

Я обнаружил, что при поиске данных с помощью alt.LookupData возвращаются последние данные. В приведенном ниже примере, когда я ищу параметр field в lookup_data для использования при определении цвета кругов, он возвращает вторую половину данных [0] * 5 + [1] * 5 (где choice_param==1).
Как мне настроить условие, чтобы использовать значения в поле, когда selection_param равно 0 (

Код: Выделить всё

selection_param==0

)?

Код: Выделить всё

import altair as alt
import numpy as np
import pandas as pd

# Generate data primary data
np.random.seed(42)
primary_data = pd.DataFrame({
'key': np.arange(10),
'x': np.random.normal(0, 1, 10),
'y': np.random.normal(0, 1, 10),
})

# Generate lookup data
lookup_data = pd.DataFrame({
'key': list(np.arange(10)) * 2,
'selection_param': [0] * 10 + [1] * 10,
'field': [0, 1, 2, 3, 4] * 2 + [0] * 5 + [1] * 5,
})

# Create chart
alt.Chart(primary_data).mark_circle().encode(
x='x',
y='y',
color='field:N'
).transform_lookup(
'key',
from_=alt.LookupData(
lookup_data,
key='key',
fields=['field']
)
)

Подробнее здесь: https://stackoverflow.com/questions/791 ... ups-altair