Schrodinger Equation, Finite Difference, Quantum Machanics, Numerical Methods, Python.
plt.figure(figsize=(12,10)) for i in range(len(z)): y = [] y = np.append(y,vec[:,z[i]]) y = np.append(y,0) y = np.insert(y,0,0) plt.plot(x,y,lw=3, label="{} ".format(i)) plt.xlabel('x', size=14) plt.ylabel('$\psi$(x)',size=14) plt.legend() plt.title('normalized wavefunctions for a harmonic