QM with Python – swing on the quantum harmonic oscillator