Func. to convert old abund. file to new format

tardis/util.py

@@ -1,8 +1,10 @@
 # Utilities for TARDIS
 
 from astropy import units as u, constants
+from pyne import nucname
 import numexpr as ne
 import numpy as np
+import pandas as pd
 import os
 import yaml
 import re
@@ -428,4 +430,19 @@ def quantity_linspace(start, stop, num, **kwargs):
         raise ValueError('Both start and stop need to be quantities with a '
                          'unit attribute')
 
-    return np.linspace(start.value, stop.to(start.unit).value, num, **kwargs) * start.unit
+    return np.linspace(start.value, stop.to(start.unit).value, num, **kwargs) * start.unit
+
+
+def convert_abundances_format(fname, delimiter='\s+'):
+    df = pd.read_csv(fname, delimiter=delimiter, comment='#', header=None)
+    #Drop shell column
+    df.drop(df.columns[0], axis=1, inplace=True)
+
+    #Creating Header rows of Element Symbol and Type of Element
+    header_df = pd.DataFrame([(nucname.name(i), 'E')
+                              for i in range(1, df.shape[1] + 1)]).transpose()
+    #Align coloumns index with abundance dataframe
+    header_df.columns = header_df.columns + 1
+
+    # Adding header rows to top and return resultant DataFrame
+    return pd.concat([header_df, df])