使用 numpy.trapz 的定积分?

Definite Integral using numpy.trapz?

我有一个函数定义为:

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?

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