Streamlit新手求助：调用session_state未定义key时出现报错

问题根源 #

你碰到的是KeyError，原因很明确：在执行st.write(st.session_state['key'])时，'key'这个键还未被添加到Streamlit的session_state中，直接访问未初始化的键就会触发报错。

修复步骤 #

按照官方文档的规范，必须在使用session_state的键之前完成初始化，以下两种写法都能解决问题：

写法一：字典格式初始化 #

# 先判断键是否存在，不存在则赋值初始值
if 'key' not in st.session_state:
    st.session_state['key'] = '初始值'

# 完成初始化后即可安全调用
st.write(st.session_state['key'])

写法二：属性格式初始化 #

# 同样先做存在性检查
if 'key' not in st.session_state:
    st.session_state.key = '初始值'

# 调用时使用属性语法
st.write(st.session_state.key)

重要提醒 #

• 初始化代码一定要放在所有使用该键的代码之前，若把初始化写在调用之后，报错问题仍会出现。
• 如果你的代码涉及多页面或交互组件，要确保初始化逻辑在组件渲染、状态访问的步骤之前执行。

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