Visualizing Quantum Mechanics with Python