- {
- “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 (2**-.5*np.exp(t*1j)*np.sin(np.pi*x)n”, “ + .5*np.exp(t*4j)*np.sin(2*np.pi*x)n”, “ + .5*np.exp(t*9j)*np.sin(3*np.pi*x))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(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”, “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
}