schrod

A simple, accurate, and fast module for solving the single particle Schrodinger equation.

Installation

1. Make sure you have python3, numpy, and scipy. All of the these (and many more) are included with anaconda.
2. Clone the repo:

git clone https://github.com/dhudsmith/schrod.git

3. From inside the schrod directory execute

python setup.py install --user

Optionally you can omit --user to install for all users.

That's it. To load the package into a python shell, simply execute

import schrod

Usage

The harmonic oscillator:

import schrod, numpy

# Specify the potential
x = numpy.linspace(-5, 5, 200)
V = 1/2 * x**2

# Create and solve Schrodinger's equation
eqn = schrod.Schrod(x, V)
eqn.solve()

# Print the first five eigenvalues
print(eqn.eigs[0:5])

Output: [ 0.5 1.5 2.50000008 3.50000122 4.50001267]

For more examples, see the examples folder.