如何用SymPy将表达式中的自定义函数替换为其定义式？

是的，完全可以用SymPy更优雅、通用地实现这个需求，而且无需手动创建符号——SymPy的sympify会自动解析并创建字符串中的所有符号。相比正则表达式，这种方法能处理复杂的嵌套表达式、运算符优先级等场景，扩展性和可读性都强得多。

实现步骤及代码 #

核心思路是：解析函数定义得到函数体，再解析目标表达式，最后用SymPy的replace方法将函数调用替换为展开后的函数体（并按需求添加括号）。

import sympy as sp
from sympy.core.parenthesized import Parenthesized

func_def = "f(x, y) = x + 2*y + 3"
expr_str = '2*a*b + f(b+a, 2*b)'

# 1. 解析函数定义为SymPy等式对象
func_eq = sp.sympify(func_def)
target_func = func_eq.lhs.func  # 获取函数对象f
func_params = func_eq.lhs.args  # 获取函数参数(x, y)
func_body = func_eq.rhs  # 获取函数体表达式x + 2*y +3

# 2. 解析目标表达式
expr = sp.sympify(expr_str)

# 3. 替换所有函数调用：将f(args)替换为带括号的展开式
result_expr = expr.replace(
    target_func,
    lambda *call_args: Parenthesized(func_body.subs(dict(zip(func_params, call_args))))
)

# 4. 转换为字符串结果
expr_eval = str(result_expr)
print(expr_eval)  # 输出: 2*a*b + ((b + a) + 2*(2*b) + 3)

为什么这比正则表达式更好？ #

• 通用性强：能处理正则难以覆盖的场景，比如嵌套函数调用（如f(g(a), h(b))）、复杂运算符组合、分数/指数等数学表达式。
• 自动符号管理：无需手动声明a、b、x等符号，sympify会自动识别并创建这些符号对象。
• 正确处理优先级：SymPy会根据数学运算符优先级自动处理括号，也可以像示例中那样通过Parenthesized强制添加外层括号。
• 可扩展性高：如果需要处理多个函数，只需重复解析函数定义的步骤，无需修改复杂的正则规则。

扩展说明 #

如果需要处理多个自定义函数，只需将每个函数定义解析为等式对象，然后依次调用replace方法即可。例如：

# 假设有另一个函数定义
g_def = "g(z) = z**2 - 1"
g_eq = sp.sympify(g_def)
expr = expr.replace(g_eq.lhs.func, lambda *args: Parenthesized(g_eq.rhs.subs(dict(zip(g_eq.lhs.args, args)))))

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