DOC: fix sqrt in axis labels (#264)

Co-authored-by: Lena Pöpping <lena.poepping@rub.de>
ComPWA · Mar 24, 2024 · d435774 · d435774
1 parent bde4df9
commit d435774
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 14 deletions.
diff --git a/docs/report/026.ipynb b/docs/report/026.ipynb
@@ -474,8 +474,8 @@
     "fig, axes = plt.subplots(figsize=(15, 6), ncols=2, sharey=True)\n",
     "ax1, ax2 = axes\n",
     "for ax in axes:\n",
-    "    ax.set_xlabel(R\"$\\mathrm{Re}(s)$\")\n",
-    "    ax.set_ylabel(R\"$\\mathrm{Im}(T)$\")\n",
+    "    ax.set_xlabel(R\"$\\mathrm{Re}\\,\\sqrt{s}$\")\n",
+    "ax1.set_ylabel(R\"$\\mathrm{Im}\\,T$\")\n",
     "\n",
     "ax1.plot(x, T1n[0].imag, label=R\"$T_\\mathrm{I}(s-0i)$\")\n",
     "ax1.plot(x, T1p[0].imag, label=R\"$T_\\mathrm{I}(s+0i)$\")\n",
@@ -627,9 +627,9 @@
     ")\n",
     "fig.update_scenes(\n",
     "    camera_center=dict(z=-0.2),\n",
-    "    xaxis_title_text=\"Re s\",\n",
-    "    yaxis_title_text=\"Im s\",\n",
-    "    zaxis_title_text=\"Im T\",\n",
+    "    xaxis_title_text=\"Re √s\",\n",
+    "    yaxis_title_text=\"Im √s\",\n",
+    "    zaxis_title_text=\"Im T(s)\",\n",
     "    zaxis_range=[-vmax, +vmax],\n",
     ")\n",
     "\n",

diff --git a/docs/report/027.ipynb b/docs/report/027.ipynb
@@ -690,9 +690,7 @@
     "T4p_res2 = T_IV_func(Zp**2, *args2)\n",
     "\n",
     "fig, axes = plt.subplots(figsize=(11, 6), ncols=4, sharey=True)\n",
-    "ax1, ax2, ax3, ax4 = axes\n",
-    "for ax in axes:\n",
-    "    ax.set_xlabel(R\"$\\mathrm{Re}\\,\\sqrt{s}$\")\n",
+    "ax1, ax2, ax3, ax4 = axes.flatten()\n",
     "\n",
     "ax1.plot(x, T1n_res1[0].imag, label=R\"$T_\\mathrm{I}(s-0i)$\")\n",
     "ax1.plot(x, T1p_res1[0].imag, label=R\"$T_\\mathrm{I}(s+0i)$\")\n",
@@ -713,7 +711,9 @@
     "\n",
     "for ax in axes:\n",
     "    ax.legend()\n",
+    "    ax.set_xlabel(R\"$\\mathrm{Re}\\,\\sqrt{s}$\")\n",
     "    ax.set_ylim(-1, +1)\n",
+    "ax1.set_ylabel(R\"$\\mathrm{Im}\\,T(s)$ (a.u.)\")\n",
     "\n",
     "fig.tight_layout()\n",
     "plt.show()"
@@ -1084,8 +1084,8 @@
     "fig.canvas.footer_visible = False\n",
     "ax1d1, ax1d2, ax2d1, ax2d2 = axes.flatten()\n",
     "\n",
-    "for ax in axes.flatten():\n",
-    "    ax.set_xlabel(R\"$\\mathrm{Re}\\,s$\")\n",
+    "for ax in axes[1]:\n",
+    "    ax.set_xlabel(R\"$\\mathrm{Re}\\,\\sqrt{s}$\")\n",
     "ax1d1.set_ylabel(\"Intensity (a.u.)\")\n",
     "ax2d1.set_ylabel(R\"$\\mathrm{Im}\\,s$\")\n",
     "\n",
@@ -1118,10 +1118,18 @@
     "            ax1d2.axvline(m0, c=R_color, ls=\"dashed\"),\n",
     "            ax2d1.axvline(m0, c=R_color, ls=\"dashed\", label=R\"$m_\\mathrm{res}$\"),\n",
     "            ax2d2.axvline(m0, c=R_color, ls=\"dashed\", label=R\"$m_\\mathrm{res}$\"),\n",
-    "            ax1d1.plot(x, T1y, c=T1_color, label=R\"$T_\\mathrm{I}$\")[0],\n",
-    "            ax1d1.plot(x, T2y, c=T2_color, label=R\"$T_\\mathrm{II}$\", ls=\"dotted\")[0],\n",
-    "            ax1d2.plot(x, T1y, c=T1_color, label=R\"$T_\\mathrm{I}$\")[0],\n",
-    "            ax1d2.plot(x, T3y, c=T3_color, label=R\"$T_\\mathrm{III}$\", ls=\"dotted\")[0],\n",
+    "            ax1d1.plot(x, T1y, c=T1_color, label=R\"$\\left|T_\\mathrm{I}\\right|^2$\")[0],\n",
+    "            ax1d1.plot(\n",
+    "                x, T2y, c=T2_color, label=R\"$\\left|T_\\mathrm{II}\\right|^2$\", ls=\"dotted\"\n",
+    "            )[0],\n",
+    "            ax1d2.plot(x, T1y, c=T1_color, label=R\"$\\left|T_\\mathrm{I}\\right|^2$\")[0],\n",
+    "            ax1d2.plot(\n",
+    "                x,\n",
+    "                T3y,\n",
+    "                c=T3_color,\n",
+    "                label=R\"$\\left|T_\\mathrm{III}\\right|^2$\",\n",
+    "                ls=\"dotted\",\n",
+    "            )[0],\n",
     "        ]\n",
     "        MESH = [\n",
     "            ax2d1.pcolormesh(X, Y, T1p_res1.imag, **style),\n",