From 9bdfda55435d74b4dd82fd9fe3f5279c7be00806 Mon Sep 17 00:00:00 2001 From: Till Raab Date: Fri, 20 Oct 2023 11:26:44 +0200 Subject: [PATCH] bf --- data/generate_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/generate_dataset.py b/data/generate_dataset.py index d8fbb26..242c652 100644 --- a/data/generate_dataset.py +++ b/data/generate_dataset.py @@ -116,9 +116,9 @@ def main(args): times_v_idx0, times_v_idx1 = np.argmin(np.abs(times_v - t0)), np.argmin(np.abs(times_v - t1)) for id_idx in range(len(fish_freq)): - ax.plot(times_v[times_v_idx0:times_v_idx1], fish_freq[id_idx][times_v_idx0:times_v_idx1], marker='.', color='k') + ax.plot(times_v[times_v_idx0:times_v_idx1], fish_freq[id_idx][times_v_idx0:times_v_idx1], marker='.', color='k', markersize=4) rise_idx_oi = np.array(rise_idx[id_idx][(rise_idx[id_idx] >= times_v_idx0) & (rise_idx[id_idx] <= times_v_idx1)], dtype=int) - ax.plot(times_v[rise_idx_oi], fish_freq[id_idx][rise_idx_oi], marker='o', color='tab:red') + ax.plot(times_v[rise_idx_oi], fish_freq[id_idx][rise_idx_oi], 'o', color='tab:red') plt.show()