Streamlit中pandas动态传入多列实现groupby分组的方法

这个需求完全可以实现，你踩的核心问题有两个：

• st.multiselect的返回值本身就是包含所有用户选中列名的列表，不需要手动按索引逐个取元素
• 你写的multiselect_var[0,1,2,3]是非法的Python索引写法，列表不支持传入元组批量取指定位置的元素，这种写法运行时会直接报错。

pandas的groupby原生支持传入列表形式的多列名完成动态多列分组，你只需要把下拉单选的列和多选返回的列列表合并成一个完整的分组列列表，直接传入groupby即可，不需要关心用户具体选了多少个多选列。

修正后可直接运行的代码 #

some_columns_df = df.loc[:,['gender','country','city','hoby','company','status']]
some_collumns = some_columns_df.columns.tolist()

select_box_var = st.selectbox("Choose X Column", some_collumns)
multiselect_var = st.multiselect("Select Columns To GroupBy", some_collumns)  

# 合并单选列和多选列，做去重处理避免用户选到重复列
group_columns = list(dict.fromkeys([select_box_var] + multiselect_var))

# 直接传入合并后的列列表即可，自动适配1~N个分组列的场景
test_g3 = df.groupby(group_columns).size().reset_index(name='count')

补充说明 #

• 代码里的dict.fromkeys是做保序去重，避免用户在多选框里选了和单选框一样的列，导致分组列重复引发不必要的异常
• 因为selectbox组件默认必有一个选中值，所以合并后的group_columns至少有1个元素，不会出现空分组列的报错；如果你后续调整了组件逻辑允许单选为空，再加个长度判断兜底即可
• 这种写法完全适配动态列场景：不管用户在多选框选1个、4个还是更多列，都能自动完成对应维度的分组统计，不需要写固定索引的硬编码逻辑。

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