saved file changed something

This commit is contained in:
weygoldt 2023-01-12 14:41:26 +01:00
parent 9821aea62e
commit 5a8643d8a0

View File

@ -217,8 +217,6 @@ def main(datapath: str) -> None:
# iterate through all fish # iterate through all fish
for i, track_id in enumerate(np.unique(ident[~np.isnan(ident)])[:2]): for i, track_id in enumerate(np.unique(ident[~np.isnan(ident)])[:2]):
# <------------------------------------------ Find best electrodes here
# <------------------------------------------ Iterate through electrodes
# get indices for time array in time window # get indices for time array in time window
window_index = np.arange(len(idx))[ window_index = np.arange(len(idx))[
(ident == track_id) & (time[idx] >= t0) & ( (ident == track_id) & (time[idx] >= t0) & (
@ -240,6 +238,8 @@ def main(datapath: str) -> None:
electrode = np.argsort(np.nanmean(powers_temp, axis=0))[-1] electrode = np.argsort(np.nanmean(powers_temp, axis=0))[-1]
# electrode = best_electrodes[0] # electrode = best_electrodes[0]
# <------------------------------------------ Iterate through electrodes
# plot spectrogram # plot spectrogram
plot_spectrogram(axs[0, i], data_oi[:, electrode], data.samplerate) plot_spectrogram(axs[0, i], data_oi[:, electrode], data.samplerate)