python代替mathematica_如何使用Python(Sympy)实现一个函数,在Wolfram Mathematica中实现与’_’和替换规则相同的功能？...

在Wolfram Mathematica中,我可以定义命名模式,其中_(称为

Blank)匹配任何表达式,然后在

replacement rule中使用匹配.

一个例子：

testexpr = p1[MM]*p2[NN] + p1[XX]*p2[MM] + p1[XX]^2;

FunTest[expr_] := Expand[expr] /. {(p1[l1_]*p2[l2_]) -> FF1[l1]*FF2[l2],

p1[l1_]^n_ -> 0, p2[l1_]^n_ -> 0}

FunTest[testexpr]

结果是FF1 [XX] FF2 [MM] FF1 [MM] FF2 [NN]

但是,我不知道如何在Python中使用sympy来做同样的事情.

import sympy as sp

p1 = sp.IndexedBase("p1")

p2 = sp.IndexedBase("p2")

FF1 = sp.IndexedBase("FF1")

FF2 = sp.IndexedBase("FF2")

MM,NN,XX=sp.symbols('MM NN XX')

SSlist=[MM,NN,XX]

testexpr=p1[MM]*p2[NN] + p1[XX]*p2[MM] + p1[XX]**2

def FunTest(expr):

expr=expr.subs([(p1[SS]*p2[SS2],FF1[SS]*FF2[SS2]) for SS in SSlist

for SS2 in SSlist]+[(p1[SS]**2,0) for SS in SSlist]+[(p2[SS]**2,0)

for SS in SSlist],simultaneous=True)

return expr

rest=FunTest(testexpr)

print(rest)

因此结果也是FF1 [MM] * FF2 [NN] FF1 [XX] * FF2 [MM].

但是我想知道是否有一种简单的方法可以使它更加通用,就像在Wolfram Mathematica中一样.如果SSlist是一个大型列表并且存在许多不同的变量,那么使用我的解决方案将难以实现.

I wonder whether there is an easy way without writing a loop over the whole list, for SS in SSlist, as in Mathematica. Can someone familiar with sympy give me any hints?

非常感谢！