\n",
" \n",
" 
\n",
"
\n",
" "
],
"text/plain": [
"Canvas(toolbar=Toolbar(toolitems=[('Home', 'Reset original view', 'home', 'home'), ('Back', 'Back to previous …"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"import numpy as np\n",
"import matplotlib.pyplot as plt\n",
"from matplotlib.animation import FuncAnimation\n",
"\n",
"fig, ax = plt.subplots(figsize=(9, 5))\n",
"\n",
"line_re, = ax.plot(x, psi_t[0].real, label=r\"$\\mathrm{Re}\\,\\psi$\")\n",
"line_im, = ax.plot(x, psi_t[0].imag, label=r\"$\\mathrm{Im}\\,\\psi$\")\n",
"line_p, = ax.plot(x, np.abs(psi_t[0])**2, label=r\"$|\\psi|^2$\")\n",
"\n",
"ax.set_xlim(x[0], x[-1])\n",
"\n",
"# Fixed vertical scale so it doesn't jump during animation\n",
"amp = np.max(np.abs(psi_t))\n",
"ax.set_ylim(-1.1*amp, 1.1*amp)\n",
"\n",
"ax.set_xlabel(\"lattice site\")\n",
"ax.set_ylabel(\"wave function\")\n",
"ax.legend()\n",
"ax.grid(True)\n",
"\n",
"title = ax.set_title(f\"t = {times[0]:.2f}\")\n",
"\n",
"def update(frame):\n",
" psi = psi_t[frame]\n",
"\n",
" line_re.set_ydata(psi.real)\n",
" line_im.set_ydata(psi.imag)\n",
" line_p.set_ydata(np.abs(psi)**2)\n",
"\n",
" title.set_text(f\"t = {times[frame]:.2f}\")\n",
"\n",
" return line_re, line_im, line_p, title\n",
"\n",
"ani = FuncAnimation(\n",
" fig,\n",
" update,\n",
" frames=len(times),\n",
" interval=5,\n",
" blit=False\n",
")\n",
"\n",
"plt.show() # using %matplotlib widget, this will display the animation in the notebook\n",
"# only for a short animation, otherwise it takes too long to render in the notebook\n",
"#plt.close(fig)\n",
"#HTML(ani.to_jshtml())"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "a5283d7b",
"metadata": {},
"outputs": [],
"source": [
"# In any cell, execute this if you want to stop the animation after a short time, otherwise it will run for a long time\n",
"# ani.event_source.stop()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "c57226b8",
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "python_env",
"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.13.15"
}
},
"nbformat": 4,
"nbformat_minor": 5
}
\n",
"