如何仅在用户做出对应选择时执行Streamlit数据库查询？

核心思路是避免提前执行所有查询函数，仅在用户选择确定后才触发对应查询，同时结合Streamlit的缓存机制减少重复查询成本。

步骤1：给查询函数添加缓存装饰器 #

用@st.cache_data（Streamlit 1.21+版本推荐）装饰你的数据库查询函数，确保同一查询只会执行一次，后续直接复用缓存结果：

import streamlit as st

@st.cache_data
def query_a():
    # 替换为实际的数据库查询逻辑
    return 你的查询结果DataFrame

@st.cache_data
def query_b():
    return 你的查询结果DataFrame

@st.cache_data
def query_c():
    return 你的查询结果DataFrame

步骤2：根据用户选择触发对应查询 #

方式1：字典映射（适合选项较多的场景） #

将函数对象而非函数执行结果存入字典，仅在用户选择后调用对应函数：

# 存储函数对象，而非立即执行查询
query_map = {
    "a": query_a,
    "b": query_b,
    "c": query_c
}

# 用户选择控件
selection = st.selectbox("make a selection", ["a", "b", "c"])

# 仅在选中后执行对应查询
df = query_map[selection]()

方式2：条件判断（适合选项较少的场景） #

直接通过if-elif-else判断选择，触发对应查询：

selection = st.selectbox("make a selection", ["a", "b", "c"])

if selection == "a":
    df = query_a()
elif selection == "b":
    df = query_b()
else:
    df = query_c()

原理说明 #

原来的代码ddict = {"a":query_a(), ...}会在字典初始化阶段立即执行所有三个查询函数，导致不必要的数据库开销。修改后，只有当用户选择确定后，才会调用对应的函数触发查询，结合缓存装饰器，相同选择不会重复执行数据库请求。

内容的提问来源于stack exchange，提问作者Daniel Tan