Use Python to study a single electron in an infinite square well and in the harmonic oscillator. Assume w = 2x x 10 (on the order of visible light) and a = 10-10 (one Angstrom) in the following.(a) In two figures, plot the potential for the infinite square well and the simple harmonic oscillator, along with the probability density of the first five stationary states at t = 0. Comment on the result. (b) In two figures plot the real and imaginary parts of the wavefunction of the n = 2 state of the infinite square well at t = 0, t = ma Anh; t = ma' 2x7, t = 3mal dah ; and f = me- Comment on the result. (c) In a single figure, plot the quantum and classical probability density of the ground state of the harmonic oscillator, using E, = (n + ;) hw and E = mw'A' with amplitude A for the classical case. Repeat the same for the first and fifth excited states in separate figures. How do quantum and classical probabilities compare with increasing energy