SymPy: Comment obtenir des valeurs à partir de l'expression

Question

J'essaye d'écrire une fonction SymPy simple.

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

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

Disons que U = 3/(7+s). Je voudrais que mon résultat soit 3*exp(-7*t), mais il retourne juste b*exp(-a*t).

Existe-t-il un moyen d'obtenir ces valeurs et de les affecter à a et b?

Answer 1

Bien sûr. Tout d'abord, mettre en place le système:

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

La méthode .match retourne un dictionnaire:

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

qui peut ensuite être passé comme argument à .subs:

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