SymPy构建表达式时如何避免子表达式提前展开以简化求导结果？

问题描述 #

在对复杂函数求导时，SymPy会自动将所有子表达式展开至基础符号层级，导致求导结果过于庞大难以处理。手动替换回原有子表达式的方法会因表达式结构调整遗漏大量简化机会。以下是第一阶段求导代码：

from sympy import *

aX, aY, aZ = symbols('aX aY aZ')
rotInc = Matrix(3,1,[aX, aY, aZ])
theta = sqrt((rotInc.T @ rotInc)[0,0])
incQuat = Quaternion.from_axis_angle(rotInc/theta, theta*2)

qX, qY, qZ, qW = symbols('qX qY qZ qW')
baseQuat = Quaternion(qW, qX, qY, qZ)
poseQuat = incQuat * baseQuat

d4 = diff(poseQuat, aX)
d4s = d4.subs({
    incQuat.a: symbols('iW'),
    incQuat.b: symbols('iX'),
    incQuat.c: symbols('iY'),
    incQuat.d: symbols('iZ'),
    theta: symbols('theta')
})

希望SymPy在构建表达式时就能保留类似cse返回的子表达式结构，仅在必要时展开，让求导结果简洁易用。

解决方案 #

1. 显式定义中间符号，提前封装子表达式

不要让SymPy自动展开子表达式，先定义代表中间变量的符号，再建立这些符号与基础变量的关系，用中间符号构建最终表达式。求导时会保留中间符号结构，最后按需展开或保留：

from sympy import *

# 定义基础符号
aX, aY, aZ = symbols('aX aY aZ')
qX, qY, qZ, qW = symbols('qX qY qZ qW')

# 定义中间变量符号
theta, iW, iX, iY, iZ = symbols('theta iW iX iY iZ')

# 建立中间变量与基础变量的关系
theta_rel = sqrt(aX**2 + aY**2 + aZ**2)
inc_quat_rel = Quaternion.from_axis_angle(Matrix([aX, aY, aZ])/theta, 2*theta)
iW_rel = inc_quat_rel.a
iX_rel = inc_quat_rel.b
iY_rel = inc_quat_rel.c
iZ_rel = inc_quat_rel.d

# 用中间变量构建最终表达式
incQuat = Quaternion(iW, iX, iY, iZ)
baseQuat = Quaternion(qW, qX, qY, qZ)
poseQuat = incQuat * baseQuat

# 对基础变量求导，结果保留中间符号
d4 = diff(poseQuat, aX)

# 按需代入关系展开，或直接保留简洁结构
d4_expanded = d4.subs({
    theta: theta_rel,
    iW: iW_rel,
    iX: iX_rel,
    iY: iY_rel,
    iZ: iZ_rel
})

2. 对求导结果使用cse提取公共子表达式

直接对庞大的求导结果调用cse，它会自动识别并提取重复子表达式，返回「子表达式列表+简化后主表达式」的结构，大幅提升可读性：

# 承接原代码中的d4
subexprs, simplified_d4 = cse(d4)

# 查看提取的子表达式
for var, expr in subexprs:
    print(f"{var} = {expr}")

# 简化后的主表达式用子表达式符号表示
print("简化后的求导结果：", simplified_d4)

你可以将提取的子表达式重新赋值为自定义符号，或按需展开部分子表达式，灵活控制结果的简洁度。

3. 用Derivative延迟求导，手动控制链式展开

如果需要精细控制展开层级，可使用Derivative保留未展开的求导形式，结合链式法则手动代入中间变量的导数，避免SymPy自动展开所有项：

# 预计算中间变量对aX的导数
d_theta_dax = diff(theta_rel, aX)
d_iW_dax = diff(iW_rel, aX)
d_iX_dax = diff(iX_rel, aX)
d_iY_dax = diff(iY_rel, aX)
d_iZ_dax = diff(iZ_rel, aX)

# 手动展开链式法则，只展开需要的部分
d4 = (diff(poseQuat, iW)*d_iW_dax + 
      diff(poseQuat, iX)*d_iX_dax + 
      diff(poseQuat, iY)*d_iY_dax + 
      diff(poseQuat, iZ)*d_iZ_dax +
      diff(poseQuat, theta)*d_theta_dax)

这种方法能完全掌控展开范围，最大化保留中间结构的简洁性。

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