Question

我正在尝试进行通用子表达式替换。

我想将x*y形式的任何表达式替换为c。

请注意，x和y可以是表达式中的任何符号。

在下面的代码中，我使用Wild实现此功能，但显然我无法使用Wild匹配符号，而不是使用子表达式 - 即使使用isinstance=[sp.Symbol]也是如此。

import sympy as sp
a, b, c = sp.symbols('a b c')

f= a*b - b

x = sp.Wild('x', isinstance=[sp.Symbol])
y = sp.Wild('y', isinstance=[sp.Symbol])

expr = f.replace(x*y, c)

这导致以下结果，

print expr
c

预期的答案是：c - b

非常感谢帮助！

谢谢！

Answer 1

isinstance不是Wild的真正论据。

您正在寻找的论点是properties。虽然它似乎没有记录，但该参数采用了一系列函数，这些函数应该对匹配的表达式保持为真。您可以使用它来匹配符号

In [7]: f= a*b - b

In [14]: y = sp.Wild('y', properties=[lambda x: isinstance(x, Symbol)])

In [15]: x = sp.Wild('x', properties=[lambda x: isinstance(x, Symbol)])

In [16]: f.replace(x*y, c)
Out[16]: -b + c

Answer 2

试试这个：

syms = sorted(set(a for a in f.atoms() if type(a) == sp.Symbol), key=str)
f.subs((s1*s2, c) for s1, s2 in it.combinations(syms, 2))

在这个例子中似乎可以正常工作：

import itertools as it

a, b, c, d = sp.symbols('a b c d')
f = 3*a**2*b*c**b+b*c**2+25+(a+b+c*a/4)/(a*c+3**(b*c))-8

syms = sorted(set(a for a in f.atoms() if type(a) == sp.Symbol), key=str)
repls = [(s1*s2, d) for s1,s2 in it.combinations(syms, 2)]
print('variables:     ' + str(syms))
print('replacements:  ' + str(repls))

expr = f.subs(repls)
print('Before:  ' + str(f))
print('After:   ' + str(expr))

Sympy：使用Wild符号替换

2 个答案: