python - 使用 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)