updating time on csv output (#818)

NOAA-OWP · Aug 6, 2024 · 5865910 · 5865910
1 parent c2cd1b4
commit 5865910
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/troute-network/troute/nhd_io.py b/src/troute-network/troute/nhd_io.py
@@ -2076,6 +2076,9 @@ def write_flowveldepth_csv_pkl(stream_output_directory, file_name,
     # Concatenate all temporary DataFrames vertically
     df = pd.concat(df_list)
     df.index.name = 'feature_id'
+
+    df['current_time'] = pd.to_datetime(df['t0']) + pd.to_timedelta(df['time'])
+    df = df[['current_time', 'flow', 'velocity', 'depth', 'nudge']]
 
     # File format handling
     file_format = file_name.split('.')[-1]