使用Python和Streamlit判断用户输入为数值或字符串的问题如何解决

• 原代码将整份列名列表传入is_numeric_dtype()做判断，该函数要求入参是pandas列/列的dtype对象，传入列表永远返回False，导致代码始终调用st.text_input，所有输入结果默认都是字符串类型。

场景1：针对单个列匹配对应输入组件 #

如果你是为指定列生成对应的搜索输入框，直接传入目标列的dtype做判断即可，修复后代码如下：

import pandas as pd
import streamlit as st
from pandas.api.types import is_numeric_dtype

# 替换为你实际要匹配的目标列名
target_col = "待匹配的列名"
st_input_update = st.number_input if is_numeric_dtype(df[target_col]) else st.text_input

with col1_search:
    regular_search_term = st_input_update("Enter Search Term")

st.number_input返回的结果本身就是数值类型，无需额外转换。

场景2：给所有列分别生成对应输入组件 #

如果你要给全表所有列都生成匹配类型的输入框，遍历每一列单独判断即可：

all_columns = df.columns.tolist()
col_search_term = {}

for col in all_columns:
    input_widget = st.number_input if is_numeric_dtype(df[col]) else st.text_input
    col_search_term[col] = input_widget(f"输入{col}的搜索值")

场景3：需要支持灵活输入（既可能是数值也可能是字符串） #

如果同一个输入框需要兼容两种类型的用户输入，可以统一用st.text_input，后续手动做类型转换即可：

with col1_search:
    regular_search_term = st.text_input("Enter Search Term")

# 手动尝试转换为数值类型
try:
    regular_search_term = float(regular_search_term)
    if regular_search_term.is_integer():
        regular_search_term = int(regular_search_term)
except ValueError:
    # 转换失败保留原字符串类型即可
    pass

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