如何基于动态列名筛选Pandas DataFrame？（Streamlit场景）

直接上可运行的代码，核心逻辑是把选中的年份和固定的PROVINCE列组成列名列表，用这个列表筛选原DataFrame：

import streamlit as st
import pandas as pd

# 替换成你自己的真实数据加载逻辑
main_df = pd.DataFrame({
    "PROVINCE": ["北京", "上海", "广东", "江苏"],
    2022: [100, 200, 300, 400],
    2023: [150, 250, 350, 450]
})

# 年份选择框，注意匹配原DataFrame的列名类型
selected_year = st.selectbox("选择年份", options=["2022", "2023"])

# 构造要保留的列列表：固定保留PROVINCE，加上选中的年份
# 若原DataFrame中年份列是整数类型，需转类型：int(selected_year)
filtered_columns = ["PROVINCE", selected_year]

# 生成筛选后的DataFrame
filtered_df = main_df[filtered_columns]

# 展示结果
st.dataframe(filtered_df)

注意事项 #

• 列名类型匹配：如果原DataFrame里的年份列是整数类型（比如列名是2022而非"2022"），要把selectbox的选中值转成整数，否则会提示找不到列。
• 检查列名拼写：确保PROVINCE列名和原DataFrame完全一致，避免拼写错误导致筛选失败。

内容的提问来源于stack exchange，提问作者Ridwan Nurzeha