python - 如何在 sympy 中插入替换

Question

我有以下内容，其中 a、b、c、d 都是 sympy 符号

p = a + b + 2*c + d
q = a + b + c + d

我想这样做:

p = p.subs(a + b + c + d, q)

并且想要得到这个:

p = q + c

但 p 保持不变。

我应该怎么做才能达到 p = q + c？

subs() 执行的匹配似乎正在寻找严格的排序，因此它没有拆分“2*c”项。

我应该使用replace()而不是subs()吗？

编辑:代码如下:

import sympy

a, b, c, d = sympy.symbols('a,b,c,d')

p = a + b + 2*c + d
q = a + b + c + d

r = p.subs(a + b + c + d, q)

print r

编辑#2

https://groups.google.com/forum/#!topic/sympy/b_Yv6s15Y0Q

我的问题与 groups.google 链接类似，只是在这种情况下 subs() 完成了它的工作。

Answer 1

首先对您的代码进行评论:由于您将 q 定义为 a+b+c+d 之和，因此您永远不会看到更改，即使它确实有效(但实际上没有)。其他有效的方法如下:

>>> p = a + b + 2*c + d
>>> q = var('q')
>>> p.extract_additively(a+b+c+d) + q
c + q

还有一个 extract_multiplicatively 方法。