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)