Get the principal quantum number as input from the user and plot the

wavefunction of the $H-$atom in $3D.$

Plot the radial wave function

Generate plots for ${}_{nlm}^2$ for $n=1,2$ and $3$ and all the values of $1$&$m.$

Hints:

You can use legendre and lagrange functions from MatLab or

scipy.interpolate.lagrange and scipy.special.legendre from python.

P$.$S$.$: Use only programming software $($preferably matlab or python$)$ for

the assignment and not online plotting software or Mathematica.