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