Streamlit 中的 Altair 绘图：如何添加图例？-Python教程-PHP中文网

Streamlit 中的 Altair 绘图：如何添加图例？

WBOY

发布： 2024-02-09 13:30:04

转载

1369人浏览过

streamlit 中的 altair 绘图：如何添加图例？

问题内容

我正在使用 streamlit，需要 altair 进行绘图（因为有可用的插值选项）。

给出这个简单的代码：

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

data = 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)

登录后复制

这会导致该图：

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

易森网络企业版

如果您是新用户，请直接将本程序的所有文件上传在任一文件夹下，Rewrite 目录下放置了伪静态规则和筛选器，可将规则添加进IIS，即可正常使用，不用进行任何设置；(可修改图片等)默认的管理员用户名、密码和验证码都是：yeesen系统默认关闭，请上传后登陆后台点击“核心管理”里操作如下:进入“配置管理”中的&ld

查看详情

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

正确答案

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

import streamlit as st
import altair as alt
import pandas as pd
alt.renderers.enable("html")

# Your data
data = pd.DataFrame({
    "x": [0, 1, 2, 3],
    "y": [0, 10, 15, 20],
    "z": [10, 8, 10, 1]
})


# Transform data to long format
data_long = pd.melt(data, id_vars=['x'], value_vars=['y', 'z'], var_name='category', value_name='y,z')

# Create an Altair chart
chart = 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)

登录后复制

输出：

以上就是Streamlit 中的 Altair 绘图：如何添加图例？的详细内容，更多请关注php中文网其它相关文章！