diff --git a/fieldExtraction/src/investment/postprocessing_funcs.py b/fieldExtraction/src/investment/postprocessing_funcs.py index 5e592ab..19aaa1c 100644 --- a/fieldExtraction/src/investment/postprocessing_funcs.py +++ b/fieldExtraction/src/investment/postprocessing_funcs.py @@ -342,11 +342,7 @@ def reorder_columns(df: pd.DataFrame, column_order: list[str]) -> pd.DataFrame: ) # Reorder columns to match column_order - ordered_columns = [col for col in column_order if col in df_copy.columns] - # Add columns in df that are not in column_order to the end - remaining_columns = [col for col in df_copy.columns if col not in column_order] - # Combine the ordered columns and remaining columns - final_column_order = ordered_columns + remaining_columns + final_column_order = [col for col in column_order if col in df_copy.columns] # Return the DataFrame with reordered columns return df_copy[final_column_order]