From 922c938fb67bb6ede7faf0b23f687c185674dcab Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 19 May 2023 07:57:34 +0200
Subject: [PATCH] save ids in trial analysis
---
trail_analysis.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/trail_analysis.py b/trail_analysis.py
index 5852da5..6324f6d 100644
--- a/trail_analysis.py
+++ b/trail_analysis.py
@@ -212,6 +212,8 @@ class Trial(object):
if not os.path.exists(os.path.join(self.base_path, self.folder, 'analysis')):
os.mkdir(os.path.join(self.base_path, self.folder, 'analysis'))
+ np.save(os.path.join(self.base_path, self.folder, 'analysis', 'ids.npy'), self.ids[::saveorder])
+
np.save(os.path.join(self.base_path, self.folder, 'analysis', 'fish_freq.npy'), self.fish_freq[::saveorder])
np.save(os.path.join(self.base_path, self.folder, 'analysis', 'fish_freq_interp.npy'), self.fish_freq_interp[::saveorder])