使用 Streamlit + Altair 绘制多系列 Dataframe

Question

我有以下DFelo_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 的原因以及接下来如何使用它，所以任何新的解释方式将不胜感激。

Answer 1

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

使用 Streamlit + Altair 绘制多系列 Dataframe

Plot multi-series Dataframe using Streamlit + Altair

python

plot

pandas

altair

streamlit