simplified csv reading

Deltares-research · Oct 25, 2024 · 835249e · 835249e
1 parent 32ce563
commit 835249e
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/examples/KWK_process.py b/examples/KWK_process.py
@@ -159,12 +159,13 @@
             file_yearmeanHW = os.path.join(dir_meas_gemHWLWwlAB,f'{station_name_dict[current_station]}_hw.txt')
             file_yearmeanLW = os.path.join(dir_meas_gemHWLWwlAB,f'{station_name_dict[current_station]}_lw.txt')
             file_yearmeanwl = os.path.join(dir_meas_gemHWLWwlAB,f'{station_name_dict[current_station]}_Z.txt')
-            yearmeanHW = pd.read_csv(file_yearmeanHW, sep='\\s+', skiprows=1, names=['datetime','values'], parse_dates=['datetime'], na_values=-999.9, index_col='datetime')
-            yearmeanLW = pd.read_csv(file_yearmeanLW, sep='\\s+', skiprows=1, names=['datetime','values'], parse_dates=['datetime'], na_values=-999.9, index_col='datetime')
-            yearmeanwl = pd.read_csv(file_yearmeanwl, sep='\\s+', skiprows=1, names=['datetime','values'], parse_dates=['datetime'], na_values=-999.9, index_col='datetime')
-            ax1.plot(yearmeanHW['values'],'+g', zorder=0)
-            ax1.plot(yearmeanLW['values'],'+g', zorder=0)
-            ax1.plot(yearmeanwl['values'],'+g',label='yearmean validation', zorder=0)
+            csv_kwargs = dict(sep='\\s+', na_values=-999.9, skiprows=1, header=None, parse_dates=[0], index_col=0)
+            yearmeanHW = pd.read_csv(file_yearmeanHW, **csv_kwargs)
+            yearmeanLW = pd.read_csv(file_yearmeanLW, **csv_kwargs)
+            yearmeanwl = pd.read_csv(file_yearmeanwl, **csv_kwargs)
+            ax1.plot(yearmeanHW[1],'+g', zorder=0)
+            ax1.plot(yearmeanLW[1],'+g', zorder=0)
+            ax1.plot(yearmeanwl[1],'+g',label='yearmean validation', zorder=0)
             ax1.legend(loc=2)
 
         fig1.savefig(os.path.join(dir_slotgem,f'kw{year_slotgem}-slotgemiddelden-{current_station}'))