🎨 Refactor species density profile calculations for improved clarity and efficiency

Author: chris-ashe

process/io/plot_proc.py

@@ -3628,60 +3628,28 @@ def plot_n_profiles(prof, demo_ranges, mfile_data, scan):
 
     # build species density profiles from electron profile and fractions
     # fracs = [fuel, alpha, protons, impurities, ions_total]
-    n_fuel, n_alpha, n_proton, n_imp, n_ions = [ne * f for f in fracs[:5]]
-
-    # convert to 1e19 m^-3 units for plotting (vectorised / less repetitive)
-    (
-        ne_plot,
-        n_fuel_plot,
-        n_alpha_plot,
-        n_proton_plot,
-        n_imp_plot,
-        n_ions_plot,
-    ) = [arr / 1e19 for arr in (ne, n_fuel, n_alpha, n_proton, n_imp, n_ions)]
+    n_fuel = ne * fracs[0]
+    n_alpha = ne * fracs[1]
+    n_proton = ne * fracs[2]
+    n_imp = ne * fracs[3]
+    n_ions = ne * fracs[4]
+
+    # convert to 1e19 m^-3 units for plotting (vectorised)
+    ne_plot = ne / 1e19
+    n_fuel_plot = n_fuel / 1e19
+    n_alpha_plot = n_alpha / 1e19
+    n_proton_plot = n_proton / 1e19
+    n_imp_plot = n_imp / 1e19
+    n_ions_plot = n_ions / 1e19
 
     prof.plot(
-        rho,
-        n_fuel_plot,
-        label=r"$n_{\text{fuel}}$",
-        color="#2ca02c",
-        linewidth=1.5,
-    )
-    prof.plot(
-        rho,
-        ne_plot,
-        label=r"$n_{e}$",
-        color="blue",
-        linewidth=1.5,
-    )
-    prof.plot(
-        rho,
-        n_alpha_plot,
-        label=r"$n_{\alpha}$",
-        color="#d62728",
-        linewidth=1.5,
-    )
-    prof.plot(
-        rho,
-        n_proton_plot,
-        label=r"$n_{p}$",
-        color="#17becf",
-        linewidth=1.5,
-    )
-    prof.plot(
-        rho,
-        n_imp_plot,
-        label=r"$n_{Z}$",
-        color="#9467bd",
-        linewidth=1.5,
-    )
-    prof.plot(
-        rho,
-        n_ions_plot,
-        label=r"$n_{i,total}$",
-        color="#ff7f0e",
-        linewidth=1.5,
+        rho, n_fuel_plot, label=r"$n_{\text{fuel}}$", color="#2ca02c", linewidth=1.5
     )
+    prof.plot(rho, ne_plot, label=r"$n_{e}$", color="blue", linewidth=1.5)
+    prof.plot(rho, n_alpha_plot, label=r"$n_{\alpha}$", color="#d62728", linewidth=1.5)
+    prof.plot(rho, n_proton_plot, label=r"$n_{p}$", color="#17becf", linewidth=1.5)
+    prof.plot(rho, n_imp_plot, label=r"$n_{Z}$", color="#9467bd", linewidth=1.5)
+    prof.plot(rho, n_ions_plot, label=r"$n_{i,total}$", color="#ff7f0e", linewidth=1.5)
 
     # make legend use multiple columns (up to 4) and place it to the right to avoid overlapping the plots
     handles, labels = prof.get_legend_handles_labels()