如何在Python中获取并转换用户输入的算术/比较条件

方法1：结合ast模块与operator模块安全解析 #

直接用eval存在安全风险（可能执行恶意代码），用ast.parse解析抽象语法树，提取操作符和操作数后通过operator模块映射执行，能有效规避风险。

示例代码：

import ast
import operator

# 预定义待验证的值列表
predefined_values = [3, 7, 10, 15]

# 操作符与operator函数的映射表
op_map = {
    ast.Eq: operator.eq,    # ==
    ast.NotEq: operator.ne, # !=
    ast.Lt: operator.lt,    # <
    ast.LtE: operator.le,   # <=
    ast.Gt: operator.gt,    # >
    ast.GtE: operator.ge,   # >=
    ast.Add: operator.add,  # +
    ast.Sub: operator.sub,  # -
    ast.Mult: operator.mul, # *
    ast.Div: operator.truediv # /
}

def check_condition(condition_str, value):
    try:
        # 解析用户输入的条件表达式，假设变量名为x
        tree = ast.parse(condition_str, mode='eval')
        expr = tree.body
        
        # 递归解析语法树节点
        def parse_node(node):
            if isinstance(node, ast.Name):
                if node.id == 'x':
                    return value
                raise ValueError(f"不支持的变量名: {node.id}")
            elif isinstance(node, ast.Constant):
                return node.value
            elif isinstance(node, ast.BinOp):
                left_val = parse_node(node.left)
                right_val = parse_node(node.right)
                op_func = op_map.get(type(node.op))
                if not op_func:
                    raise ValueError(f"不支持的操作符: {type(node.op).__name__}")
                return op_func(left_val, right_val)
            elif isinstance(node, ast.Compare):
                left_val = parse_node(node.left)
                for op, comparator in zip(node.ops, node.comparators):
                    right_val = parse_node(comparator)
                    op_func = op_map.get(type(op))
                    if not op_func:
                        raise ValueError(f"不支持的比较操作符: {type(op).__name__}")
                    if not op_func(left_val, right_val):
                        return False
                return True
            else:
                raise ValueError(f"不支持的表达式类型: {type(node).__name__}")
        
        return parse_node(expr)
    except Exception as e:
        print(f"条件解析错误: {e}")
        return False

# 获取用户输入的条件
user_condition = input("请输入条件（变量用x表示，例如x > 5 或 x + 3 == 10）: ")

# 遍历验证每个预定义值
for val in predefined_values:
    result = check_condition(user_condition, val)
    print(f"x = {val} 时，条件[{user_condition}]的验证结果: {result}")

方法2：简化版（使用eval，仅适合信任输入场景） #

如果能确保用户输入绝对安全（比如内部工具使用），可以直接用eval实现，代码更简洁：

predefined_values = [3, 7, 10, 15]

user_condition = input("请输入条件（变量用x表示）: ")

for x in predefined_values:
    try:
        result = eval(user_condition)
        print(f"x = {x} 时，条件[{user_condition}]的验证结果: {result}")
    except Exception as e:
        print(f"条件执行错误: {e}")

注意事项 #

• 方法1安全性更高，严格限制了支持的变量和操作符，避免恶意输入；方法2仅适合可控场景，不要用于公开访问的系统。
• 若需要支持多变量验证，可扩展parse_node函数中的变量处理逻辑，添加对应变量的映射。

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