The Python code
The code sets up the Hamiltonian matrix by defining the the minimun and maximum values of \( r \) with a
maximum value of integration points. These are set in the initialization function. It plots the
eigenfunctions of the three lowest eigenstates.