From ba3803bb56d9a3a571010a5a96f48352cf41e507 Mon Sep 17 00:00:00 2001 From: wendtalexander Date: Wed, 18 Jan 2023 11:43:11 +0100 Subject: [PATCH] =?UTF-8?q?patrick=20=C3=BCbernimmt?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- code/chirpdetection.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/code/chirpdetection.py b/code/chirpdetection.py index 3e6e0c1..3272900 100644 --- a/code/chirpdetection.py +++ b/code/chirpdetection.py @@ -12,7 +12,7 @@ from sklearn.preprocessing import normalize from modules.filters import bandpass_filter, envelope, highpass_filter from modules.filehandling import ConfLoader, LoadData from modules.plotstyle import PlotStyle -from modules.timestamps import group_timestamps, group_timestamps_v2 + ps = PlotStyle() @@ -649,10 +649,20 @@ def main(datapath: str) -> None: t0 = (3 * 60 * 60 + 6 * 60 + 43.5) data_oi = data.raw[window_starts[0]:window_starts[-1]+ int(dt*data.raw_rate), 10] plot_spectrogram(ax, data_oi, data.raw_rate, t0) - chirps = np.concatenate(chirps) - for ch in chirps: + chirps_concat = np.concatenate(chirps) + for ch in chirps_concat: ax. axvline(ch, color='b', lw=1) + chirps_new = [] + chirps_ids = [] + [chirps[x] for x in tr_index] + for tr in np.unique(fish_ids): + tr_index = np.asarray(fish_ids) == tr + + ts = list(np.ravel(chirps[fish_ids == int(tr)])) + chirps_new.extend(ts) + chirps_ids.extend(list(np.ones_like(ts)*tr)) +