update(PrmsData): update load_from_file method for more robust parsin…

…g of data file
pygsflow · Apr 21, 2023 · 11600d7 · 11600d7
1 parent 7c183a0
commit 11600d7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gsflow/prms/prms_data.py b/gsflow/prms/prms_data.py
@@ -88,7 +88,11 @@ def load_from_file(data_file):
                     columns.append(val_nm[0] + "_" + str(val))
 
         columns = GsConstant.COLUMN_HEADER + columns
-        data_pd = pd.read_csv(fid, delim_whitespace=True, names=columns)
+        data_pd = pd.read_csv(fid, delim_whitespace=True, header=None)
+        if len(list(data_pd)) > len(columns):
+            data_pd = data_pd[[i for i in range(len(list(columns)))]]
+        data_pd.rename(columns={ix: i for ix, i in enumerate(columns)}, inplace=True)
+
         Dates = []
         for index, irow in data_pd.iterrows():
             dt = datetime.datetime(