Contents:
Plot absolute square of wave function
E (np.ndarray) – Eigenenergies
psi (np.ndarray) – Eigenfunctions