Python中如何将字符串形式的集合运算表达式转换为可执行运算并解决未预定义集合变量的报错问题

核心问题原因 #

• 直接调用eval未指定自定义命名空间时，表达式里的A、B等变量没有提前定义，触发name 'A' not defined报错
• 拆分字符转set的方案逻辑错误：把表达式拆成['A','&','B']后直接给字符套set()，相当于把变量名本身当成集合内容，或者用默认空集填充，完全没用到用户实际输入的自然数集合值，结果必然不符合预期
• Python内置set原生支持&（交集）、|（并集）、-（差集）、^（对称差）运算符，不需要额外转换运算符格式

可行实现方案 #

核心思路：先从用户输入的表达式中提取所有用到的集合变量，再引导用户输入每个变量对应的集合元素，最后把这些「变量-集合」对作为自定义命名空间传给eval，既解决未定义报错，又避免全局变量污染。

完整可运行代码 #

import re

# 接收用户输入的集合运算表达式
expr = input("请输入集合运算表达式（如A&B|C）：").strip()

# 提取表达式中的所有集合变量：默认匹配单大写字母，适配示例里A/B/C的格式
# 若需要匹配多字符变量（如Set1|Set2），可把正则替换为 r'[A-Za-z_][A-Za-z0-9_]*'
variables = sorted(set(re.findall(r'[A-Z]', expr)))

# 逐个获取用户输入的每个集合变量的元素
var_dict = {}
for var in variables:
    elems = input(f"请输入集合{var}的元素，用逗号分隔（如4,5）：").strip()
    # 转换为自然数set，空输入则生成空集
    var_dict[var] = set(map(int, elems.split(','))) if elems else set()

# 传入自定义命名空间执行eval：第二个空字典清空全局权限，避免安全风险
result = eval(expr, {}, var_dict)

print(f"运算结果：{result}")

运行示例 #

输入运算表达式：A&B
输入集合A元素：4,5
输入集合B元素：5,6
输出结果：{5}

可选优化点 #

• 增加输入校验：判断用户输入的元素是否为自然数，避免类型错误
• 支持括号运算：上述代码原生兼容Python set的括号优先级逻辑，不需要额外调整
• 若担心eval的安全风险，可改用ast模块解析表达式做安全校验，普通本地使用场景下上述代码足够稳定

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