Sympy subs函数无法替换可识别Subs实例问题排查

问题重现 #

你遇到的问题是：用同一符号替换两个不同的Subs实例（分别为eta(s)和eta(sbar)在0点的一阶导数），分两次调用subs后仍有一个Subs实例未被替换；但用不同符号替换时则正常。复现代码及现象如下：

复现代码 #

from sympy import Function, symbols, Symbol, expand

eta = Function('eta')
s, c, h = symbols('s c h', real=True)
sbar, cbar, hbar = symbols('sbar cbar hbar', real=True)

Number_of_derivatives = 5

eta_dsymbol = []
eta_ds = []
eta_dsbar = []

for ix in range(0, Number_of_derivatives + 1):
    eta_ds.append(eta(s).diff(s, ix).subs(s, 0))
    eta_dsbar.append(eta(sbar).diff(sbar, ix).subs(sbar, 0))
    eta_dsymbol.append(Symbol("eta_d" + str(ix), real=True))

替换现象 #

ix = 1
Subs1 = eta_dsbar[ix]
Subs2 = eta_ds[ix]
sym = eta_dsymbol[ix]

eq = ((eta(s).diff(s,1).subs(s,0)+1)**2 *
      (eta(sbar).diff(sbar,1).subs(sbar,0)+2)**3 *
      (eta(s).diff(s,1).subs(s,0)+eta(sbar).diff(sbar,1).subs(sbar,0)+5)**7).expand()

# 两次同一符号替换后仍有Subs残留
eq.subs(Subs1, sym).subs(Subs2, sym)
# 输出包含未替换的Subs(Derivative(eta(s), s), s, 0)
_.has(Subs)  # 返回True

# 不同符号替换则正常
A = Symbol("A")
eq.subs(Subs1, sym).subs(Subs2, A).has(Subs)  # 返回False
eq.subs(Subs1, A).subs(Subs2, sym).has(Subs)  # 返回False

原因解析 #

1. Subs实例的唯一性：Subs1（eta(sbar)的导数在0点）和Subs2（eta(s)的导数在0点）是两个完全独立的Sympy对象，因为它们的原变量分别是sbar和s。Sympy内部通过对象的全部参数（包括原变量）判断是否相等，因此不会将二者视为同一个符号。
2. 连续subs的匹配干扰：分两次调用subs且替换目标为同一符号时，第一次替换会修改表达式树结构，将所有Subs1替换为sym。第二次替换Subs2为sym时，表达式中已存在的sym节点会干扰匹配逻辑——部分Subs2实例可能与sym形成复合项（如乘积、幂次），导致精确匹配无法识别这些节点；或是内部缓存机制误判了匹配优先级，造成遗漏。
3. 不同符号替换无干扰：使用不同符号替换时，两次替换的目标完全独立，不会出现表达式树中已有符号与待替换符号的混淆，因此所有Subs实例都能被精确匹配并替换。

解决办法 #

避免分两次调用subs，而是一次性传递包含所有替换规则的字典，让Sympy一次性遍历表达式树完成所有替换：

eq.subs({Subs1: sym, Subs2: sym}).has(Subs)  # 返回False，所有Subs均被替换

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