使用 Streamlit + Altair 绘制多系列数据框

如何解决使用 Streamlit + Altair 绘制多系列数据框

我有以下 DF elo_df

date        Person1 Person2 Person3 Person4 Person5 Person6                     
2020-12-31  1500    1500    1500    1500    1500    1500
2021-01-01  1480    1506    1506    1500    1506    1500
2021-01-06  1513    1495    1495    1490    1515    1490
2021-01-08  1506    1502    1502    1490    1508    1490

我想使用 altair 绘制它并使其可缩放和交互。此外，我希望能够将 Y 域设置为 [1350,1600]。我有一个 players_names=["Person1",...,"Person6"] 的列表，我想使用它，以防多一个人。

这是我目前所做的

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

[...part where I obtain elo_df...]

k = alt.Chart(elo_df.reset_index()).transform_fold(
    players_names).mark_line().encode(
        alt.X('date'),alt.Y('value',scale=alt.Scale(domain=[1350,1600])),color='variable',)

    st.altair_chart(k)

这导致我出现以下错误： ValueError: variable encoding field is specified without a type; the type cannot be inferred because it does not match any column in the data.

即使在阅读了文档之后，我也没有完全弄清楚 transform_fold 的原因以及接下来如何使用它，因此将不胜感激任何新的解释方式。

解决方法

Altair 使用 Pandas 来确定绘制数据时要使用的数据类型。对于不属于 Pandas 数据帧的任何数据，例如来自 URL 的数据或来自转换的列，您需要explicitly specify what type the data is：

import altair as alt
import pandas as pd


elo_df = pd.read_clipboard()
players_names=elo_df.filter(like='Person').columns.tolist()
chart = alt.Chart(elo_df.reset_index()).transform_fold(
    players_names).mark_line().encode(
        alt.X('date:T'),alt.Y('value:Q',scale=alt.Scale(domain=[1350,1600])),color='key:N',)

chart