Question

我正在尝试使用[SymPy] [1]同时替换表达式中的多个术语。我用字典作为参数尝试了[subs function] [2]，但发现它依次替换。

In : a.subs({a:b, b:c})
Out: c

问题是第一次替换导致一个术语可以被第二次替换取代，但它不应该（因为我的原因）。

有关如何同时进行替换的任何想法，而不会相互干扰吗？

修改： 这是一个真实的例子

In [1]: I_x, I_y, I_z = Symbol("I_x"), Symbol("I_y"), Symbol("I_z")

In [2]: S_x, S_y, S_z = Symbol("S_x"), Symbol("S_y"), Symbol("S_z")

In [3]: J_is = Symbol("J_IS")

In [4]: t = Symbol("t")

In [5]: substitutions = (
(2 * I_x * S_z, 2 * I_x * S_z * cos(2 * pi * J_is * t) + I_y * sin(2 * pi * J_is * t)),
(I_x,  I_x * cos(2 * pi * J_is * t) + 2 * I_x * S_z * sin(2 * pi * J_is * t)),
(I_y,  I_y * cos(2 * pi * J_is * t) - 2 * I_x * S_z * sin(2 * pi * J_is * t))
)

In [6]: (2 * I_x * S_z).subs(substitutions)
Out[7]: (I_y*cos(2*pi*J_IS*t) - 2*I_x*S_z*sin(2*pi*J_IS*t))*sin(2*pi*J_IS*t) + 2*S_z*(I_x*cos(2*pi*J_IS*t) + 2*I_x*S_z*sin(2*pi*J_IS*t))*cos(2*pi*J_IS*t)

只应进行适当的替换，在这种情况下只会发生第一次替换。所以预期的输出应该如下：

In [6]: (2 * I_x * S_z).subs(substitutions)
Out[7]: I_y*sin(2*pi*J_IS*t) + 2*I_x*S_z*cos(2*pi*J_IS*t)

Answer 1

当前版本的sympy提供关键字同步。以前的答案中的复杂操作不再是必要的：

In [1]: (x*sin(y)).subs([(x,y),(y,x)],simultaneous=True)
Out[1]: y⋅sin(x)

Answer 2

subs(self,*args)方法以这种方式定义（部分）：

In [11]: x.subs??
...
sequence = args[0]
if isinstance(sequence, dict):
    return self._subs_dict(sequence)
elif isinstance(sequence, (list, tuple)):
    return self._subs_list(sequence)

如果你通过subs dict，你就会失去对替换顺序的控制权。如果您传递subs列表或元组，则可以控制订单。

这不允许您同时进行替换。如果用户传递x.subs([(x,y),(y,x)])之类的内容，那将导致困难。所以我怀疑sympy有一种同时进行替换的方法。相反，我相信所有的变量都是无序的（如果你传递一个字典），或者最好是通过1遍有序替换（如果你传递一个列表或元组）：

In [17]: x.subs([(x,y),(y,z)])
Out[18]: z

In [19]: x.subs([(y,z),(x,y)])
Out[19]: y

PS。 _subs_list(self, sequence)的定义（部分）如下：

In [14]: x._subs_list??
...
    for old, new in sequence:
        result = result.subs(old, new)

这确定了潜艇的完成顺序。

Answer 3

@~unutbu's answer的示例：

>>> import ordereddict # collections.OrderedDict in Python 2.7+
>>> from sympy import *
>>> x,y,z = symbols('xyz')
>>> x.subs(ordereddict.OrderedDict([(x,y),(y,z)]))
y
>>> x.subs(ordereddict.OrderedDict([(y,z),(x,y)]))
z

Answer 4

回答编辑过的问题。

在您的示例中，您可以使用一些临时变量，这些变量在后续替换时不会被覆盖。然后，一旦完成所有可能重叠的替换，就可以用真实替换替换临时变量。

此示例适用于该问题，如果您的完整问题包含更复杂的替换，我认为您仍应该能够创建临时变量以避免重叠替换。

from sympy import Symbol, sin, cos, pi

I_x, I_y, I_z = Symbol("I_x"), Symbol("I_y"), Symbol("I_z")
S_x, S_y, S_z = Symbol("S_x"), Symbol("S_y"), Symbol("S_z")
J_is = Symbol("J_IS")
t = Symbol("t")
I_x_temp, I_y_temp, I_z_temp = Symbol("I_x_temp"), Symbol("I_y_temp"), Symbol("I_z_temp")

f = 2*I_x*S_z
answer = I_y*sin(2*pi*J_is*t) + 2*I_x*S_z*cos(2*pi*J_is*t)

subs1a = [
    (2*I_x*S_z, 2*I_x_temp*S_z*cos(2*pi*J_is*t) + I_y_temp*sin(2*pi*J_is*t)),
    (I_x,  I_x_temp*cos(2* pi*J_is*t) + 2*I_x_temp*S_z*sin(2*pi*J_is*t)),
    (I_y,  I_y_temp*cos(2*pi*J_is*t) - 2*I_x_temp*S_z* sin(2*pi*J_is*t))
]

subs_temp = [(I_x_temp, I_x), (I_y_temp, I_y), (I_z_temp, I_z)]

print f
f = f.subs(subs1a)
print f
f = f.subs(subs_temp)
print f
print f == answer # True

注意，您也可以背靠背执行两次替换：

f.subs(subs1a).subs(subs_temp) == answer

Answer 5

无论输入（字典或序列）如何，关键字simultaneous都会执行非冲突的潜艇：

>>> x.subs([(x,y),(y,z)],simultaneous=1)
y
>>> x.subs([(y,z),(x,y)],simultaneous=1)
y

SymPy中的非顺序替换

5 个答案: