在Altair（Python）中实现世界地图和滑块。

Question

我想要使用Python的Altair库构建一个互动世界地图图表，显示每个国家和日期的COVID-19确诊病例数，并使用日期滑块进行操作。

数据格式：

"country_region","date","confirmed_cases"
"Afghanistan",2020-01-22,0
"Afghanistan",2020-01-23,0
"Afghanistan",2020-01-24,0
"Afghanistan",2020-01-25,0

我已经成功地构建了一个非交互式图表，但是无法设置滑块。我的图表现在看起来像这样：

我想添加一个滑块来选择日期，但是无法正确设置它。

我使用transform_lookup方法根据数据（确诊病例/人口）为国家添加颜色，transform_filter应该被添加以在用户改变滑块位置时执行数据过滤。但是这不起作用。我的假设是，在transform_lookup中始终使用原始数据，因此过滤器无法工作。我没有找到有关同时使用transform_lookup和滑块的示例或文档。

非常感谢听到任何可以帮助我解决这个问题的想法。

代码:

import requests
import json

import pandas as pd

import altair as alt
from vega_datasets import data
from altair import datum

df = pd.read_csv('https://raw.githubusercontent.com/MariaKokshaikina/any-data/main/covid19_global_confirmed_cases%20(1).csv')

country_info = requests.get(
    'https://raw.githubusercontent.com/MariaKokshaikina/any-data/main/country_info.json'
).json()

df = df[df['country_region'].isin(country_info)]
df = df.sort_values('date', ascending=True)
df = df.tail(5000)

def timestamp(t):
    return pd.to_datetime(t).timestamp() * 1000

df['id'] = df['country_region'].map(lambda x: country_info[x]['numericCode'])
df['rate'] = df['confirmed_cases'] / df['country_region'].map(lambda x: country_info[x]['population'])
df['timestamp'] = df['date'].map(timestamp)

countries = alt.topo_feature(data.world_110m.url, 'countries')

slider = alt.binding_range(
    step=24 * 60 * 60 * 1000,
    min=df['timestamp'].min(), 
    max=df['timestamp'].max()
)

select_date = alt.selection_single(
    name="slider", 
    fields=['timestamp'],
    bind=slider, 
)

alt.Chart(countries).mark_geoshape()\
    .encode(color='rate:Q')\
    .add_selection(select_date)\
    .transform_filter(select_date)\
    .transform_lookup(
        lookup='id',
        from_=alt.LookupData(df, key='id', fields=['rate'])
    )\
    .project('equirectangular')\
    .properties(
        width=500,
        height=300,
        title='Title'
    )

• 您可以删除.transform_filter(select_date)行，以查看带有滑块的工作地图，而该滑块不会更改任何内容。

- maria

Answer 1

问题似乎在于在 Vega 中，查找转换不会根据选择动态重新计算。您可以通过切换主要数据源来解决此问题，以便所有时间戳都出现在最终连接的数据集中：

alt.Chart(df).mark_geoshape()\
    .encode(color='rate:Q')\
    .add_selection(select_date)\
    .transform_filter(select_date)\
    .transform_lookup(
        lookup='id',
        from_=alt.LookupData(countries, key='id',
                             fields=["type", "properties", "geometry"])
    )\
    .project('equirectangular')\
    .properties(
        width=500,
        height=300,
        title='Title'
    )

- jakevdp