Python中如何将用户输入的字符串函数转为可执行函数？

方法1：使用lambda + eval（简单直接，适合自用场景） #

因为你是自己编写工具，eval的安全风险可以忽略，这个方法最快实现。核心思路是把用户输入的字符串表达式替换掉大写X为小写x，再用eval生成可执行函数：

import math

# 获取用户输入
func_str = input("请输入函数表达式（如2X+1）：").strip()
# 处理大写X转小写x，同时确保math模块的函数能被识别
func_str = func_str.replace('X', 'x')
# 生成可执行函数
try:
    f = lambda x: eval(func_str, {"__builtins__": None}, {"x": x, **math.__dict__})
    # 测试执行
    print(f(2))  # 输入2X+1的话，这里会输出5
except Exception as e:
    print(f"函数解析错误：{e}")

说明：把math的所有方法传入eval的命名空间，用户输入sin(X)、cos(X)这类含数学函数的表达式也能正常执行。

方法2：使用ast模块（更安全，适合对外工具） #

如果后续要把工具给其他人用，担心eval的安全问题，可以用ast模块解析表达式，只允许合法的数学运算，避免执行恶意代码：

import ast
import math

def create_function(func_str):
    # 替换大写X为小写x
    func_str = func_str.replace('X', 'x')
    # 解析表达式为AST语法树
    try:
        tree = ast.parse(func_str, mode='eval')
    except SyntaxError as e:
        raise ValueError(f"无效的表达式：{e}")
    
    # 遍历AST节点，限制仅允许使用x变量和math模块函数
    for node in ast.walk(tree):
        if isinstance(node, ast.Name) and node.id not in ['x'] + dir(math):
            raise ValueError(f"不允许使用的变量/函数：{node.id}")
    
    # 编译并生成可执行函数
    code = compile(tree, filename='<ast>', mode='eval')
    def f(x):
        return eval(code, {"__builtins__": None}, {"x": x, **math.__dict__})
    return f

# 使用示例
try:
    func_str = input("请输入函数表达式：").strip()
    f = create_function(func_str)
    print(f(3))  # 输入2X+1的话输出7
except Exception as e:
    print(f"错误：{e}")

说明：通过AST节点检查，过滤掉非法变量和函数，大幅降低安全风险。

额外注意事项 #

• 统一大小写：把用户输入的所有大写X转为小写x，避免因大小写不一致导致解析失败
• 支持复杂表达式：上述方法兼容X**2（平方）、X/2、log(X)这类常见数学表达式
• 异常捕获：必须加try-except块，在用户输入非法表达式时给出清晰的错误提示

内容的提问来源于stack exchange，提问作者Tomás