aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorThéo de la Hogue2022-11-16 20:00:25 +0100
committerThéo de la Hogue2022-11-16 20:00:25 +0100
commit980abf0392021296a6c3b5f765277cd9b74f64a2 (patch)
tree4ab0108af07dc702a57b773a63b309bc0ded6b90
parent16622e12a903ea0c311838af2bcee385a950d320 (diff)
downloadargaze-980abf0392021296a6c3b5f765277cd9b74f64a2.zip
argaze-980abf0392021296a6c3b5f765277cd9b74f64a2.tar.gz
argaze-980abf0392021296a6c3b5f765277cd9b74f64a2.tar.bz2
argaze-980abf0392021296a6c3b5f765277cd9b74f64a2.tar.xz
Exporting to csv using pandas method directly.
-rw-r--r--src/argaze/utils/tobii_segment_aruco_aoi_export.py2
-rw-r--r--src/argaze/utils/tobii_segment_gaze_movements_export.py6
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