如何用等式左侧部分替换表达式对应部分？SymPy技术咨询

嘿，我刚好碰到过类似的需求，SymPy提供了几种简单的方式来实现这个替换，让你的结果显示成c²而不是(a+b)²，下面给你一步步演示：

方法1：使用subs()直接替换 #

这是最直观的方法，直接把表达式里的a + b替换成c就行，代码如下：

from sympy import *
from sympy.abc import *
init_printing()

c = a + b
d = (1/3)*c**3
e = d.diff(a)

# 执行替换
e_simplified = e.subs(a + b, c)
display(e_simplified)  # 会显示 c²

subs()可以接受单个替换对，或者字典形式的多个替换规则，非常灵活。

方法2：使用replace()进行模式匹配替换 #

如果你的表达式更复杂，需要匹配特定的模式，replace()会更合适，它支持基于SymPy表达式的模式匹配：

# 接上面的代码
e_simplified = e.replace(a + b, c)
display(e_simplified)

这个方法和subs()在这个场景下效果一致，但如果后续你需要替换更复杂的子表达式（比如带系数的、特定结构的），replace()的模式匹配能力会更有用。

方法3：结合等式对象进行替换 #

如果你已经用Eq定义了c和a+b的关系，可以用xreplace()来批量替换：

# 先定义等式关系
eq = Eq(c, a + b)
# 把等式右侧的a+b替换成左侧的c
e_simplified = e.xreplace({eq.rhs: eq.lhs})
display(e_simplified)

这种方式的好处是可以统一管理变量间的等式关系，适合有多个替换规则的场景。

小提示 #

如果你希望在求导过程中就尽可能保留c的形式，其实可以在定义d的时候，确保SymPy把c当作一个独立符号处理？不过实际上求导时SymPy会自动展开c的定义，所以求导后替换是最可靠的方式。

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