{
“cells”: [
{

“cell_type”: “markdown”, “metadata”: {}, “source”: [

“# Square Well”

]

}, {

“cell_type”: “code”, “execution_count”: 1, “metadata”: {}, “outputs”: [

{
“data”: {
“text/plain”: [

“<Figure size 640x480 with 3 Axes>”

]

}, “metadata”: {}, “output_type”: “display_data”

}

], “source”: [

“import numpy as npn”, “import matplotlib.pyplot as pltn”, “from matplotlib.animation import PillowWritern”, “import animatplot as ampn”, “n”, “n”, “def psi(x, t):n”, “ return (n”, “ 2**-0.5 * np.exp(t * 1j) * np.sin(np.pi * x)n”, “ + 0.5 * np.exp(t * 4j) * np.sin(2 * np.pi * x)n”, “ + 0.5 * np.exp(t * 9j) * np.sin(3 * np.pi * x)n”, “ )n”, “n”, “n”, “x = np.linspace(0, 1, 20)n”, “t = np.linspace(0, 10, 20)n”, “n”, “X, T = np.meshgrid(x, t)n”, “Y1 = psi(X, T).realn”, “Y2 = psi(X, T).imagn”, “n”, “timeline = amp.Timeline(t, "s", 24)n”, “n”, “ax = plt.axes(xlim=[0, 1], ylim=[-2, 2])n”, “block1 = amp.blocks.Line(X, Y1, ax=ax)n”, “block2 = amp.blocks.Line(X, Y2, ax=ax)n”, “n”, “anim = amp.Animation([block1, block2], timeline)n”, “n”, “# Your standard matplotlib stuffn”, “plt.title(n”, “ r"Particle in a Box: $|\Psi\rangle = \frac{1}{\sqrt{2}}"n”, “ r"|E_1\rangle + \frac{1}{2}|E_2\rangle + \frac{1}{2}|E_3\rangle$",n”, “ y=1.03,n”, “)n”, “plt.xlabel("position")n”, “plt.ylabel(r"$\Psi$")n”, “plt.legend(["Real", "Imaginary"])n”, “n”, “anim.toggle()n”, “anim.timeline_slider()n”, “n”, “anim.save("sq_well.gif", writer=PillowWriter(fps=5))n”, “plt.show()”

]

}, {

“cell_type”: “raw”, “metadata”: {

“raw_mimetype”: “text/restructuredtext”

}, “source”: [

“.. image:: sq_well.gif”

]

}

], “metadata”: {

“celltoolbar”: “Raw Cell Format”, “kernelspec”: {

“display_name”: “Python 3”, “language”: “python”, “name”: “python3”

}, “language_info”: {

“codemirror_mode”: {

“name”: “ipython”, “version”: 3

}, “file_extension”: “.py”, “mimetype”: “text/x-python”, “name”: “python”, “nbconvert_exporter”: “python”, “pygments_lexer”: “ipython3”, “version”: “3.6.5”

}

}, “nbformat”: 4, “nbformat_minor”: 2

}