diff --git a/Trajectories/plot_traj.py b/Trajectories/plot_traj.py index 7b98f87c77bbceae7980436d52ec9a6d47b82df4..897cb46a177cff59c23832b892bfdc8cd6766c5a 100755 --- a/Trajectories/plot_traj.py +++ b/Trajectories/plot_traj.py @@ -88,13 +88,9 @@ if __name__ == "__main__": else: # args.min_duration > 1 n_long_traj = 0 + duration_array = get_duration(expanded_traj, e_overestim) - for traj in expanded_traj: - duration = report_graph.node_to_date_eddy(traj[- 1], e_overestim, - only_date = True) \ - - report_graph.node_to_date_eddy(traj[0], e_overestim, - only_date = True) - + for traj, duration in zip(expanded_traj, duration_array): if duration >= args.min_duration: n_long_traj += 1 plot_single_traj(traj, e_overestim, SHPC, args.orientation, ax,