dataframe trim done

Author: Saurabh Pandey

PostProcessing/dataframes.py

@@ -5,6 +5,16 @@
 def repair_dropped_frames(df: pd.DataFrame) -> pd.DataFrame:
     pass
 
+def save_dataframes(dataframes, prefix='df'):
+    # Generate filenames based on a pattern or numbering scheme
+    filenames = [f"{prefix}{i}.csv" for i in range(1, len(dataframes) + 1)]
+
+    # Save each DataFrame to a separate file
+    for i, df in enumerate(dataframes):
+        filename = filenames[i]
+        df.to_csv(filename, index=False, header=False)
+    print("DataFrames saved successfully.")
+
 
 # Function to find the largest value in the first entry of all dataframes
 def find_largest_first_entry(dfs):
@@ -39,10 +49,14 @@ def compute_time_range(dfs) -> Tuple[int, int]:
 # Function to trim dataframes based on specified values
 def trim_into_interval(dfs, min_common, max_common, threshold):
     trimmed_dataframes = []
-    # import pdb;pdb.set_trace()
     for df in dfs:
-        start: pd.DataFrame = df[(df.iloc[:, 0] >= min_common - threshold) & (df.iloc[:, 0] <= min_common + threshold)]
-        end: pd.DataFrame = df[(df.iloc[:, 0] >= max_common - threshold) & (df.iloc[:, 0] <= max_common + threshold)]
-        trimmed_df = df[(df.iloc[:, 0] >= start.iloc[0, 0]) & (df.iloc[:, 0] <= end.iloc[0, 0])].reset_index(drop=True)
-        trimmed_dataframes.append(trimmed_df)
+        start: pd.DataFrame = df[df.iloc[:, 0].between(min_common-threshold, min_common+threshold, inclusive='both')]
+        end: pd.DataFrame = df[df.iloc[:, 0].between(max_common-threshold, max_common+threshold, inclusive='both')]
+        if not start.empty and not end.empty :
+            df_start = start.stack().iloc[0]
+            df_end = end.stack().iloc[-1]
+            trimmed_df = df[df.iloc[:, 0].between(df_start, df_end, inclusive='both')]
+            trimmed_dataframes.append(trimmed_df)
+        else:
+            print("No values found within the specified range.")
     return trimmed_dataframes