A module for solving and visualizing the Schrödinger equation
QMsolve: A module for solving and visualizing the Schrödinger equation
QMsolve seeks to provide easy solid and easy to use solver, capable of solving the Schrödinger equation for one and two particles, and creating descriptive and stunning visualizations of its solutions both in 1D, 2D, and 3D.
pip install qmsolve
How it works
The way this simulator works is by discretizing and Hamiltonian of an arbitrary potential,
specified as a function of the particle observables. This is achieved with the
Hamiltonian.solve method efficiently diagonalizes the Hamiltonian and outputs the energies and the eigenstates of the system.
Finally, the eigenstates can be plotted with the
visualization.superpositions method features the possibility of interactively visualizing a superposition of the computed eigenstates and studying
the time dependence of the resulting wavefunction.
For efficiently solving the time dependent Schrödinger equation,
TimeSimulation class must be used. It takes as an argument the Hamiltonian you have previously defined, and the method you desire to use.
For a quick start, take a look to the examples found in the examples subdirectory.
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.