用Python作图求薛定谔方程

Question

我试着画出有限势阱薛定谔方程的解。它是一个超越方程 f（E）=2 cot（（2mE）^1/2L/h）+（（U-E）/E^1/2-（E/（U-E））^1/2。在

其中E是变量

我的代码给了我一个参差不齐的图。在

import pylab as pi
import pylab as p
import numpy as np


L = 10e-9 # meters
U = 1.602e-20 # joules
m = 10e-9 # kilograms
h = 1.055e-34 # meters squared * kilograms / seconds


x = p.linspace(0, 1.601999e-20, 1000) # before negative in square root
y = 2/p.tan(2*p.pi*((2*m*x)**(1/2))*(L/h)) + ((U-x)/x)**1/2 - (x/(U-x))**1/2


tol = 100
y[y > tol] = np.nan
y[y < -tol] = np.nan


p.plot(x, y, 'g-', lw=1)
p.show()

有人能帮忙找出问题吗？在