Ensure alpha symbol used in plot labels

WMD-group · Feb 19, 2024 · 74b0183 · 74b0183
1 parent d32a59a
commit 74b0183
Show file tree

Hide file tree

Showing 2 changed files with 86 additions and 86 deletions.
diff --git a/examples/PyTASER_DFT_Example.ipynb b/examples/PyTASER_DFT_Example.ipynb
diff --git a/pytaser/plotter.py b/pytaser/plotter.py
@@ -348,14 +348,14 @@ def _rescale_overlapping_curves(list_of_curves):
                     plt.plot(
                         energy_mesh[xmin_ind:xmax_ind],
                         normalised_tas[xmin_ind:xmax_ind],
-                        label="Total TAS (Δα)",
+                        label=r"Total TAS ($\Delta\alpha$)",
                         color="black",
                         lw=3.5,
                         alpha=0.75,  # make semi-transparent to show if overlapping lines
                     )
 
                 else:  # yaxis = "alpha"
-                    abs_label = "a (a.u.)"
+                    abs_label = r"$\alpha$ (a.u.)"
                     transition_dict = {}  # control transition plotting here rather than later on:
                     alpha_normalisation_factor = np.max(  # normalise to max alpha in dark
                         np.abs(self.alpha_dark)[xmin_ind:xmax_ind]
@@ -367,15 +367,15 @@ def _rescale_overlapping_curves(list_of_curves):
                             - self.alpha_light_dict["emission"][xmin_ind:xmax_ind]
                         )
                         / alpha_normalisation_factor,
-                        label="a (light)",
+                        label=r"$\alpha$ (light)",
                         color="black",
                         lw=2.5,
                         alpha=0.75,  # make semi-transparent to show if overlapping lines
                     )
                     plt.plot(
                         energy_mesh[xmin_ind:xmax_ind],
                         self.alpha_dark[xmin_ind:xmax_ind] / alpha_normalisation_factor,
-                        label="a (dark)",
+                        label=r"$\alpha$ (dark)",
                         color="blue",
                         lw=2.5,
                         alpha=0.75,  # make semi-transparent to show if overlapping lines
@@ -470,7 +470,7 @@ def _rescale_overlapping_curves(list_of_curves):
                 plt.plot(
                     energy_mesh[xmin_ind:xmax_ind],
                     jdos_diff[xmin_ind:xmax_ind],
-                    label="Total TAS (ΔJDOS only)",
+                    label="Total TAS ($\\Delta$JDOS only)",
                     color="black",
                     lw=3.5,
                     alpha=0.75,  # make semi-transparent to show if overlapping lines