Streamlit中实现交互式DataFrame更新：选中行后隐藏已选类别并解决重复显示问题

我明白你想要实现的交互式DataFrame功能——让用户逐轮选择行，每次选完对应类别的所有行就隐藏，直到选够指定数量的类别，同时解决现在每次操作后DataFrame重复显示的问题。咱们来调整你的代码，解决这些问题：

首先，原代码里的while循环不太适合Streamlit的rerun机制，会导致重复渲染；另外session_state的使用没有完全跟上状态变化，这是重复显示的核心原因。下面是修正后的完整代码：

import streamlit as st
import pandas as pd

# 初始化DataFrame
d = {'Num': [1, 2, 3, 4, 5, 6, 7, 8, 9],
     'types': ["A", "A", "A", "B", "B", "B", "C", "C", "C"]}
original_df = pd.DataFrame(data=d)
original_df['selected'] = None

# 初始化session_state，保存核心状态
if 'df' not in st.session_state:
    st.session_state.df = original_df.copy()
if 'selected_categories' not in st.session_state:
    st.session_state.selected_categories = []

# 目标选择数量（可根据需求修改，比如30）
TARGET_SELECTIONS = 3

# 先检查是否已经选够了
if len(st.session_state.selected_categories) < TARGET_SELECTIONS:
    # 计算当前可选的类别：排除已选的
    selectable_categories = [cat for cat in original_df['types'].unique() if cat not in st.session_state.selected_categories]
    # 过滤出只显示可选类别的DataFrame
    df_to_display = st.session_state.df[st.session_state.df['types'].isin(selectable_categories)]

    # 渲染可交互的DataFrame
    select_dict = st.dataframe(
        df_to_display,
        use_container_width=True,
        hide_index=True,
        on_select="rerun",
        selection_mode="multi-row",
    )

    # 处理用户选中的行
    if select_dict.selection.rows:
        selected_rows = select_dict.selection.rows
        # 获取选中行对应的类别
        selected_cat = df_to_display.iloc[selected_rows]['types'].unique()[0]
        # 更新已选类别列表
        if selected_cat not in st.session_state.selected_categories:
            st.session_state.selected_categories.append(selected_cat)
        # 标记该类别下的所有行为已选中
        st.session_state.df.loc[st.session_state.df['types'] == selected_cat, 'selected'] = True
        # 手动触发rerun，更新界面
        st.rerun()
else:
    st.success(f"已完成{TARGET_SELECTIONS}个类别的选择！")
    st.dataframe(st.session_state.df, use_container_width=True, hide_index=True)

关键修改点说明： #

• 去掉while循环，改用session_state跟踪状态：Streamlit是每次用户交互都会重新运行整个脚本，while循环会导致多次重复渲染，改用session_state保存已选类别列表，自然控制流程。
• 统一基于session_state维护DataFrame：所有对DataFrame的修改都直接操作st.session_state.df，确保状态在rerun之间正确保留，不会每次都重置回原始DataFrame。
• 一次只渲染一个DataFrame：每次脚本运行时，只会根据当前状态渲染可选的DataFrame或者完成提示，避免重复显示。
• 按类别批量标记选中：用户选中某类别下的任意行后，直接标记该类别所有行为已选中，符合你“隐藏已选类别所有选项”的需求，不用逐行处理合并。

这样调整后，用户每次选中某类别下的行，页面就会自动刷新，隐藏该类别所有行，只显示剩余可选类别，直到选够目标数量，而且不会出现重复显示DataFrame的问题，完全适配多轮选择的场景（哪怕是30个类别也能轻松扩展，只要修改TARGET_SELECTIONS的值就行）。

备注：内容来源于stack exchange，提问作者Claudia Behnke