SymPy中如何将复合表达式整体替换为单个变量？解决部分替换问题

问题核心：直接使用subs替换x+y为z时，SymPy默认的自动展开机制会破坏部分表达式结构，导致替换不彻底。比如示例中2*(x+y)被展开为2x+2y，无法被subs(x+y, z)匹配，仅(x+y)**2部分完成替换。

解决方案1：创建表达式时禁用自动展开 #

在初始化Eq对象时添加evaluate=False参数，阻止SymPy自动展开表达式，这样subs就能匹配所有x+y的实例：

from sympy import *
x, y, z = symbols('x y z')

# 禁用自动展开，保留原式结构
eq = Eq(2*(x+y) + 3*(x+y)**2, 0, evaluate=False)
eq2 = eq.subs(x+y, z)
print(eq2)  # 输出：Eq(2*z + 3*z**2, 0)

解决方案2：对已展开的表达式先收集公因子 #

如果表达式已经被展开，先用collect方法将x+y作为公因子整合，再执行替换：

from sympy import *
x, y, z = symbols('x y z')

eq = Eq(2*(x+y) + 3*(x+y)**2, 0)
# 收集x+y的公因子，还原结构
collected_lhs = collect(eq.lhs, x+y)
eq2 = Eq(collected_lhs, eq.rhs).subs(x+y, z)
print(eq2)  # 输出：Eq(2*z + 3*z**2, 0)

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