SymPy:如何从表达式中获取值

时间:2012-02-18 17:20:03

标签: python pattern-matching substitution sympy

我正在尝试编写一个简单的SymPy函数。

a = Wild('a')
b = Wild('b')
p = Wild('p')
q = Wild('q')
...

if (U).match(b/(a+s)):
    return b*exp(-a*t)

让我们说U = 3/(7+s)。我希望我的结果为3*exp(-7*t),但它只返回b*exp(-a*t)

有没有办法获取这些值并将它们分配给a和b?

1 个答案:

答案 0 :(得分:1)

不确定。首先,设置系统:

>>> from sympy import var, Wild, exp
>>> s = var("s")
>>> t = var("t")
>>> a = Wild("a")
>>> b = Wild("b")
>>> U = 3/(7+s)

.match方法返回字典:

>>> U.match(b/(a+s))
{b_: 3, a_: 7}
>>> m = U.match(b/(a+s))

然后可以作为参数传递给.subs

>>> target = b*exp(-a*t)
>>> target
b_*exp(-t*a_)
>>> target.subs(m)
3*exp(-7*t)