PHP前端开发

Streamlit 中的 Altair 绘图:如何添加图例?

百变鹏仔 1天前 #Python
文章标签 图例
问题内容

我正在使用 streamlit,需要 altair 进行绘图(因为有可用的插值选项)。

给出这个简单的代码:

import streamlit as stimport altair as altimport pandas as pddata = pd.DataFrame({"x": [0, 1, 2, 3], "y": [0, 10, 15, 20], "z": [10, 8, 10, 1]})base = alt.Chart(data.reset_index()).encode(x="x")chart = alt.layer(    base.mark_line(color="red").encode(y="y", color=alt.value("green")),    base.mark_line(color="red").encode(y="z", color=alt.value("red")),).properties(title="My plot",)st.altair_chart(chart, theme="streamlit", use_container_width=True)

这会导致该图:

在图旁边添加图例的正确方法是什么?

在文档中,我将图例选项视为“颜色”的一部分,但这似乎始终与可视化另一个维度有关。就我而言,我只想绘制不同的线条并用它们各自的颜色绘制图例。


正确答案


将您的数据转换为长数据帧格式。这种格式更适合在 altair 中创建图例,因为每行都与一个类别相关联。然后使用这个类别进行颜色编码:

import streamlit as stimport altair as altimport pandas as pdalt.renderers.enable("html")# Your datadata = pd.DataFrame({    "x": [0, 1, 2, 3],    "y": [0, 10, 15, 20],    "z": [10, 8, 10, 1]})# Transform data to long formatdata_long = pd.melt(data, id_vars=['x'], value_vars=['y', 'z'], var_name='category', value_name='y,z')# Create an Altair chartchart = alt.Chart(data_long).mark_line().encode(    x='x',    y='y,z',    color='category:N'  # Use the category field for color encoding).properties(    title="My plot")st.altair_chart(chart, use_container_width=True)

输出: