column_names: Define balmer_names0, pass into gauss_lines_names0 (iss #…

…25)
astrochun · Feb 22, 2020 · ad0b5fd · ad0b5fd
1 parent e8b6c5c
commit ad0b5fd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Metallicity_Stack_Commons/column_names.py b/Metallicity_Stack_Commons/column_names.py
@@ -1,4 +1,4 @@
-from . import line_name
+from . import line_name, line_type
 
 # These are common/general column names
 
@@ -20,9 +20,12 @@
 # This is just the suffix
 gauss_names0 = ['Flux_Gaussian', 'Flux_Observed', 'S/N', 'Center', 'Norm',
                 'Median', 'Sigma']
+balmer_names0 = ['Abs_Norm', 'Abs_Sigma']
 gauss_lines_names0 = []
-for line0 in line_name:
+for line0, type0 in zip(line_name, line_type):
     gauss_lines_names0 += ['{}_{}'.format(line0, suffix) for suffix in gauss_names0]
+    if type0 == 'Balmer':
+        gauss_lines_names0 += ['{}_{}'.format(line0, suffix) for suffix in balmer_names0]
 
 # Temperature and metallicity properties
 temp_metal_names0 = ['T_e', '12+log(O/H)', 'log(O+/H)', 'log(O++/H)', 'O+/H', 'O++/H']