- #1
Milpool
- 9
- 0
Homework Statement
I'm currently working on a project in which I have to solve the energy eigenvalues of the Schrodinger equation to compute the mass of certain Mesons. We've been taught very little programming (so apologies that my understanding is very basic), and are therefore given any stock programs we will need, for example my Runge-Kutta. I need to define a function that guesses a value of energy, solves my equation and spits out the solution for r=0, so that I can minimise this to satisfy the boundary condition of the problem that u(r=0)=0. I cannot seem to work out how to return u(0) from my Runge-Kutta program. Below is the runkut program and my definition of my Schrodinger equation.
Python:
def derivs(n, x, y):
"Defines Schrodinger equation"
dy=zeros(n+1,float)
if x==0: y[2]==0 #otherwise divide by zero error
else:
dy[1]=y[2]
dy[2]=(-E+x+((L)/x**2))*y[1]
return dy
def runkut(n, x, y, h):
"Advances the solution of diff eqn defined by derivs from x to x+h"
y0=y[:]
k1=derivs(n, x, y)
for i in range(1,n+1): y[i]=y0[i]+0.5*h*k1[i]
k2=derivs(n, x+0.5*h, y)
for i in range(1,n+1): y[i]=y0[i]+h*(0.2071067811*k1[i]+0.2928932188*k2[i])
k3=derivs(n, x+0.5*h, y)
for i in range(1,n+1): y[i]=y0[i]-h*(0.7071067811*k2[i]-1.7071067811*k3[i])
k4=derivs(n, x+h, y)
for i in range(1,n+1):
a=k1[i]+0.5857864376*k2[i]+3.4142135623*k3[i]+k4[i]
y[i]=y0[i]+0.16666666667*h*a
x+=h
return (x,y)
N=150
N0=34
L=2.0
E=1.0 #introduce loop
xmax=25.0
xmin=0.0
dx=(xmax-xmin)/N
Homework Equations
-
The Attempt at a Solution
I have attempted to print values of x and y[1] for a range of x as a test
Python:
x=0.0; y=[0, 0.0, 0.0]
z=[0 for j in range(0,N)]
for j in range(0,N):
(x,y) = runkut(2,x,y,dx)
print x, y[1]
z[j]=y[1]
As a note once I have a functioning code returning u(0), am then comfortable as to where to go from there, it is just wrapping my head around how to simply return u(0). Any help would be hugely appreciated, thanks.
Last edited: