如何安全处理pandas query中str.contains的输入,避免报错并支持反斜杠搜索?
解决方案
直接避开df.query的字符串拼接方式,改用df.loc结合str.contains的参数化调用,彻底解决特殊字符转义和语法错误问题:
# 获取用户输入 user_input = input() # 生成匹配掩码,用regex=False做字面匹配(避免正则转义干扰) match_mask = df[field_name].str.contains(user_input, case=False, regex=False) # 筛选结果 filtered_df = df.loc[match_mask]
关键说明
- 避免字符串拼接风险:原代码用f-string拼接查询语句,一旦输入包含三引号、反斜杠这类特殊字符,会直接破坏查询语句的语法结构导致报错。改用参数化传递输入值后,pandas会自动处理字符串的转义逻辑,无需手动处理特殊字符。
- 字面匹配反斜杠:添加
regex=False参数后,str.contains会把用户输入的内容当作字面文本匹配,比如输入a\s时,会精准匹配单元格中包含a\s的内容(像aaaa\saaaaa这类行就会被正确筛选出来)。如果需要启用正则匹配,只需去掉regex=False即可,但此时要注意正则语法的转义规则。
内容的提问来源于stack exchange,提问作者Mario
相关产品推荐
相关产品推荐

