使用numpy.trapz的定積分？

Question

我已經定義了一個功能：

def probability(x,t,i): 
    return np.real(np.conjugate(TD_Psi(x,t,i))*TD_Psi(x,t,i)) 

我到目前爲止已經使用整合的以下方法在Python

for t in range(0,10): 
      PD = (np.trapz(probability(x,t,initial_state),x)) 
      print (PD) 

有沒有什麼辦法可以調整這在集成的某些值x例如從x = 0到x = 5？

Answer 1

當您將x數組傳遞給np.trapz函數時，指定積分範圍。爲了從0到5進行積分，例如使用該範圍內的1000個函數值，應該這樣做：

x = np.linspace(0,5,1000) 
for t in range(0,10): 
    PD = (np.trapz(probability(x,t,initial_state),x)) 
    print (PD)