如何让SymPy将计算结果以自定义高阶表达式形式精简显示

这个问题我之前处理类似表达式时也遇到过，SymPy默认的subs做精确匹配，所以当表达式被展开后（比如2x+2y），直接用subs(x+y, z)确实识别不出它是2*(x+y)。不过有几个实用的方法能实现你想要的效果，让结果自动用定义的高阶符号（比如z=x+y）来表示：

方法1：使用Wild通配符匹配表达式结构 #

Wild可以帮你匹配任意形式的子表达式结构，不管它有没有被展开。我们可以用它来匹配x+y的倍数形式，替换成对应的z的倍数：

from sympy import *
x, y, z = symbols('x, y, z')
f = z**2
F = Matrix([f.subs(z, x+y)]).jacobian(Matrix([x, y]))

# 定义一个通配符w，用来匹配x+y的系数
w = Wild('w')
# 匹配"系数*(x+y)"的结构，替换为"系数*z"
F_replaced = F.replace(w*(x + y), w*z)
pprint(F_replaced)

运行后会输出：

[2*z  2*z]

这个方法的优势是灵活，你可以根据需要定义更复杂的匹配规则，比如匹配a*x + b*y（当a=b时）替换成a*z，或者其他自定义结构。

方法2：用公共子表达式提取（CSE）自动替换 #

如果你的雅可比矩阵里有很多重复的子表达式，用sympy.cse可以自动提取这些公共部分，再替换成你想要的符号，适合处理更复杂的场景：

from sympy import *
x, y, z = symbols('x, y, z')
f = z**2
F = Matrix([f.subs(z, x+y)]).jacobian(Matrix([x, y]))

# 提取公共子表达式
substitutions, simplified_exprs = cse(F)
# 把提取出的x+y替换成z
new_substitutions = []
for var, expr in substitutions:
    if expr == x + y:
        new_substitutions.append((var, z))
    else:
        new_substitutions.append((var, expr))
# 重新构建表达式
final_F = Matrix([expr.subs(new_substitutions) for expr in simplified_exprs])
pprint(final_F)

这个方法会自动找出所有重复的子表达式，你只需要指定要替换的目标（比如x+y→z）即可，非常适合大规模的复杂表达式处理。

为什么直接subs会失效？ #

SymPy的subs默认是精确结构匹配，当2x+2y被展开后，它和x+y已经不是完全相同的表达式结构了，所以subs(x+y, z)无法识别到隐含的倍数关系。而replace和cse是基于表达式的结构特征或公共子表达式来匹配，所以能绕过展开的问题。

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