Sympy：用式发言替代表达式

Question

我正在使用Sympy（在Sagemath中）。我想用刹车通用（用于量子机械问题）进行替代。下面有一个简约的代码，以重现问题。

from sympy.physics.quantum import Bra, Ket
from sympy import *
theta=symbols('theta',commutative=True)
pi, mu= symbols("pi mu",commutative=False)
W=2*pi*mu
print(W.subs(pi*mu,theta))
V=Bra(pi)*Ket(mu)
print(V.subs(Bra(pi)*Ket(mu),theta))
U=2*Bra(pi)*Ket(mu)
print(U.subs(Bra(pi)*Ket(mu),theta))

输出是：

2*theta
theta
2*<pi|*|mu>

如果没有领先的标量乘数，则替换可以很好地工作。我被更复杂的表达困住了。

Answer 1

下

srepr(U)
# out: "Mul(Integer(2), Bra(Symbol('pi', commutative=False)), Ket(Symbol('mu', commutative=False)))"

srepr(Bra(pi)*Ket(mu))
# out: "InnerProduct(Bra(Symbol('pi', commutative=False)),Ket(Symbol('mu', commutative=False)))"

在这些情况 。使用命令U.Subs（bra（pi）*ket（mu），theta）您要求搜索一个类型innerproduct的对象u /code>，但没有，因此没有替代。

在这种情况下，您必须执行：

U.subs(Mul(Bra(pi), Ket(mu)),theta)
# out: 2*theta

编辑：或正如@oscar Benjamin指出的那样，您要这样做：

from sympy.physics.quantum import qapply
U = qapply(U)
srepr(U)
# out: "Mul(Integer(2), InnerProduct(Bra(Symbol('pi', commutative=False)),Ket(Symbol('mu', commutative=False))))"

现在，您可以看到innerproduct作为mul的参数。最后：

U.subs(Bra(pi)*Ket(mu), theta)
# out: 2*theta