'H-atom wavefunctions'

DPotoyan · Nov 8, 2023 · b276b16 · b276b16
1 parent 4294510
commit b276b16
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ch05/hydrogen_wavefunctions.ipynb b/ch05/hydrogen_wavefunctions.ipynb
@@ -42108,10 +42108,14 @@
       "outputs": [],
       "source": [
         "def plot_orbital_2d(n,l,m, max_r = 10):\n",
+        "    '''Adjut orbital extent max_r to fit the plots'''\n",
+        "\n",
         "    fig, ax = plt.subplots()\n",
         "  \n",
         "    im = ax.contour(normalized_wavefunction(n, l, m, max_r=max_r), 16, cmap='rocket', extent=[-max_r, max_r,-max_r, max_r])\n",
+        "\n",
         "    ax.set_title(r'$|\\psi_{{({0}, {1}, {2})}}|^2$'.format(n, l, m), fontsize=15)\n",
+        "    \n",
         "    ax.set_aspect('equal')"
       ]
     },