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)