From 980abf0392021296a6c3b5f765277cd9b74f64a2 Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Wed, 16 Nov 2022 20:00:25 +0100 Subject: Exporting to csv using pandas method directly. --- src/argaze/utils/tobii_segment_aruco_aoi_export.py | 2 +- src/argaze/utils/tobii_segment_gaze_movements_export.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/argaze/utils/tobii_segment_aruco_aoi_export.py b/src/argaze/utils/tobii_segment_aruco_aoi_export.py index 4a866e5..0e9e302 100644 --- a/src/argaze/utils/tobii_segment_aruco_aoi_export.py +++ b/src/argaze/utils/tobii_segment_aruco_aoi_export.py @@ -330,7 +330,7 @@ def main(): print(f'Markers {marker_id} has been detected in {tracked_count} / {try_count} frames ({round(100 * tracked_count / try_count, 2)} %)') # Export aruco aoi data - ts_offset_aois.save_as_csv(vs_data_filepath) + ts_offset_aois.as_dataframe().to_csv(vs_data_filepath, index=True) print(f'Aruco AOI data saved into {vs_data_filepath}') # Notify when the aruco aoi video has been exported diff --git a/src/argaze/utils/tobii_segment_gaze_movements_export.py b/src/argaze/utils/tobii_segment_gaze_movements_export.py index fdf4a91..fee8960 100644 --- a/src/argaze/utils/tobii_segment_gaze_movements_export.py +++ b/src/argaze/utils/tobii_segment_gaze_movements_export.py @@ -155,15 +155,15 @@ def main(): print(f'\n{len(fixations)} fixations and {len(saccades)} saccades found') # Export fixations analysis - fixations.save_as_csv(fixations_filepath) + fixations.as_dataframe().to_csv(fixations_filepath, index=True) print(f'Fixations saved into {fixations_filepath}') # Export saccades analysis - saccades.save_as_csv(saccades_filepath) + saccades.as_dataframe().to_csv(saccades_filepath, index=True) print(f'Saccades saved into {saccades_filepath}') # Export gaze status analysis - gaze_status.save_as_csv(gaze_status_filepath) + gaze_status.as_dataframe().to_csv(gaze_status_filepath, index=True) print(f'Gaze status saved into {gaze_status_filepath}') # Prepare video exportation at the same format than segment video -- cgit v1.1