temp_metallicity_calc.metallicity_calculation: Use column_names temp_…

…metal_names0 convention (iss #21, #25)
astrochun · Mar 15, 2020 · daa8199 · daa8199
1 parent f07d244
commit daa8199
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Metallicity_Stack_Commons/temp_metallicity_calc.py b/Metallicity_Stack_Commons/temp_metallicity_calc.py
@@ -1,6 +1,7 @@
 import numpy as np
 
 from . import k_dict
+from .column_names import temp_metal_names0
 
 k_4363 = k_dict['OIII_4363']
 k_5007 = k_dict['OIII_5007']
@@ -73,7 +74,8 @@ def metallicity_calculation(T_e, TWO_BETA, THREE_BETA):
     com_O = O_s_ion + O_d_ion
     com_O_log = np.log10(com_O) + 12
 
-    metal_dict = dict(O_s_ion=O_s_ion, O_d_ion=O_d_ion,
-                      O_s_ion_log=O_s_ion_log, O_d_ion_log=O_d_ion_log)
+    keys = temp_metal_names0[2:]
+    arr0 = [O_s_ion, O_d_ion, O_s_ion_log, O_d_ion_log]
+    metal_dict = dict(zip(keys, arr0))
 
     return com_O_log, metal_dict