python - SymPy: 式から値を取得する方法

Question

シンプルな 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 に割り当てる方法はありますか?

Answer 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)