将Math.js风格表达式转换为可运行Python代码

核心问题分析 #

从math.js获取的表达式存在两个Python不兼容的语法点：

• 用^表示幂运算，而Python中^是按位异或，正确的幂运算符是**
• 允许省略乘号（如5x），但Python必须显式写*

之前用sympify只返回计算结果是因为用了subs代入求值，而非生成可复用的函数；eval也会因为语法不兼容报错。

解决方案步骤 #

1. 表达式预处理 #

先把math.js语法转成Python兼容的表达式，用正则表达式处理常见情况：

import re

def preprocess_mathjs_expr(expr):
    # 替换幂运算符^为Python的**
    expr = expr.replace('^', '**')
    # 补全省略的乘号：数字后接变量（如5x→5*x）、括号后接变量（如(x)x→(x)*x）
    expr = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', expr)
    expr = re.sub(r'(\))([a-zA-Z])', r'\1*\2', expr)
    return expr

2. 生成可调用的Python函数 #

有两种常用方式：

方式一：用SymPy的lambdify生成函数

适合需要符号运算支持的场景，生成的函数性能较好：

from sympy import symbols, sympify, lambdify

# 定义变量
x = symbols('x')
# 输入math.js表达式
mathjs_expr = "2*x^2+5x-1"
# 预处理表达式
processed_expr = preprocess_mathjs_expr(mathjs_expr)
# 转换为SymPy符号表达式
sym_expr = sympify(processed_expr)
# 生成Python可调用函数
f = lambdify(x, sym_expr)

# 测试调用
print(f(2))  # 输出17

方式二：通过字符串拼接+exec生成函数

适合不需要SymPy依赖的轻量场景：

def create_python_func(expr):
    processed_expr = preprocess_mathjs_expr(expr)
    # 拼接函数代码字符串
    func_code = f"""def f(x):
    return {processed_expr}"""
    # 执行代码并获取函数
    local_ns = {}
    exec(func_code, globals(), local_ns)
    return local_ns['f']

# 使用示例
f = create_python_func("2*x^2+5x-1")
print(f(2))  # 输出17

注意事项 #

• 如果表达式中有多个变量（如x、y），只需调整预处理和函数生成逻辑，比如定义多个symbols，或者生成带多个参数的函数
• 若处理不可信来源的表达式，exec和eval存在安全风险，优先使用SymPy的方案，避免执行恶意代码

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