adding final plot to script

This commit is contained in:
wendtalexander 2023-01-18 11:23:47 +01:00
parent ef61cec695
commit 1cdf813b6d

View File

@ -637,15 +637,19 @@ def main(datapath: str) -> None:
""" """
bool_vector[cm] = False bool_vector[cm] = False
chirps.append(the_real_chirps) chirps.append(the_real_chirps)
fish_ids.append(track_id)
for ct in the_real_chirps: for ct in the_real_chirps:
axs[0, el].axvline(ct, color='b', lw=1) axs[0, el].axvline(ct, color='b', lw=1)
plt.close()
embed() embed()
fig, ax = plt.subplots() fig, ax = plt.subplots()
t0 = (3 * 60 * 60 + 6 * 60 + 43.5) t0 = (3 * 60 * 60 + 6 * 60 + 43.5)
data_oi = data.raw[window_starts[0]:window_starts[-1]+ int(dt*data.raw_rate), 10] 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) plot_spectrogram(ax, data_oi, data.raw_rate, t0)
chirps = np.concatenate(chirps)
for ch in chirps: for ch in chirps:
ax. axvline(ch, color='b', lw=1) ax. axvline(ch, color='b', lw=1)