Sympy的subs无法替换子表达式，如何实现MATLAB同款效果？

SymPy自带的subs方法仅能处理简单场景的子表达式替换，无法覆盖所有复杂情况，但MATLAB可以完美完成这类替换任务。以下是在SymPy中实现等效效果的方法：

示例对比 #

SymPy 现有替换效果 #

MATLAB 目标替换效果 #

解决方案 #

1. 使用replace()方法替代subs #

replace()是SymPy中专门用于子表达式匹配替换的方法，支持更复杂的结构匹配，基础用法如下：

from sympy import symbols, replace

x, y, r = symbols('x y r')
expr = x**2 + y**2 + 3*(x**2 + y**2)
# 直接匹配并替换指定子表达式
result = replace(expr, x**2 + y**2, r**2)

2. 结合通配符实现批量匹配 #

如果需要匹配带系数的同类子表达式，可以用Wild定义通配符，实现类似MATLAB的批量替换：

from sympy import symbols, Wild, replace

x, y, r = symbols('x y r')
a = Wild('a')  # 定义通配符匹配任意系数或前置表达式
expr = x**2 + y**2 + 3*(x**2 + y**2)
# 匹配任意结构下的目标子表达式并替换
result = replace(expr, a*(x**2 + y**2), a*r**2)

3. 处理嵌套复杂表达式 #

对于嵌套层级较深的表达式，replace()默认会深度遍历所有子表达式进行匹配；如果需要精准匹配特定层级，可通过deep=False参数限制匹配范围。

内容的提问来源于stack exchange，提问作者Александр Алексеев