求解谐波势阱中粒子的Shrodinger方程

时间:2019-10-24 01:54:47

标签: python numpy

你好(这是我第一次在堆栈溢出中发帖),我正在尝试使用射手法计算谐波势中粒子的前三个能级

该代码改编自Mark Newman的《计算物理》中的一个脚本,该脚本计算了盒子中粒子的基态。

这是链接http://www-personal.umich.edu/~mejn/cp/programs/squarewell.py

这是我改编的代码

import numpy as np

from scipy.constants import m_e,hbar,elementary_charge

#define constants

vo = 50
a = 1e-11

#define limits of integration for adaptive runge-kutta method

xi = -10*a
xf = 10*a
N = 1000
dx = (xf-xi)/N

def V(x):#define harmonic potential well
    return vo*((x**2)/(a**2))

def f(r,x,E): #schrodinger's equation
    psi = r[0]
    phi = r[1]
    fpsi = psi
    fphi = (2*m_e/(hbar**2))*(V(x)-E)*psi
    return np.array([fpsi,fphi],float)

def solve(E): #calculates wave function for an energy E
    psi = 0.0
    phi = 1.0
    r = np.array([psi,phi],float)
    for x in np.arange(xi,xf,dx): #adaptive runge-kutta method
        k1 = dx*f(r,x,E)
        k2 = dx*f(r+0.5*k1,x+0.5*dx,E)
        k3 = dx*f(r+0.5*k2,x+0.5*dx,E)
        k4 = dx*f(r+k3,x+dx,E)
        r += (k1+2*k2+2*k3+k4)/6
    return r[0]

#finds the energy using secant method

E1 = 0.0
E2 = elementary_charge
psi2 = solve(E1)
target = elementary_charge/1000

while abs(E1-E2)>target:
    psi1,psi2 = psi2,solve(E2)
    E1,E2 = E2,E2-psi2*(E2-E1)/(psi2-psi1)
    print (E2/elementary_charge)

运行时出现此错误

RuntimeWarning: invalid value encountered in double_scalars

E1,E2 = E2,E2-psi2*(E2-E1)/(psi2-psi1)

我认为这意味着psi2和psi1距离太近,但我不确定如何解决此问题

1 个答案:

答案 0 :(得分:2)

是的!您认为值彼此之间太接近是正确的。您的代码返回一个nan。这是因为被零除。

我建议使用校正因子。分母中的max(delta, (psi2-psi1))之类的内容,其中delta仍然可以是很小的值,但会阻止division by zero