diff --git a/Trajectories/plot_traj.py b/Trajectories/plot_traj.py index 897cb46a177cff59c23832b892bfdc8cd6766c5a..ecf12b5463148fad3015c6cc5ca3b5aaab3e6927 100755 --- a/Trajectories/plot_traj.py +++ b/Trajectories/plot_traj.py @@ -26,9 +26,9 @@ def get_extr_coord(traj, e_overestim, SHPC, orientation): return x, y def plot_single_traj(traj, e_overestim, SHPC, orientation, ax, src_crs, - annotate_flag): + annotate_flag, color): x, y = get_extr_coord(traj, e_overestim, SHPC, orientation) - ax.plot(x, y, color = "red", linewidth = 0.5, transform = src_crs) + ax.plot(x, y, color, linewidth = 0.5, transform = src_crs) ax.plot(x[0], y[0], marker = "s", markersize = 2, color = "black", transform = src_crs) @@ -84,7 +84,7 @@ if __name__ == "__main__": if args.min_duration == 1: for traj in expanded_traj: plot_single_traj(traj, e_overestim, SHPC, args.orientation, ax, - src_crs, args.annotate) + src_crs, args.annotate, color = "red") else: # args.min_duration > 1 n_long_traj = 0 @@ -94,7 +94,7 @@ if __name__ == "__main__": if duration >= args.min_duration: n_long_traj += 1 plot_single_traj(traj, e_overestim, SHPC, args.orientation, ax, - src_crs, args.annotate) + src_crs, args.annotate, color = "red") print("Number of trajectories with sufficient duration:", n_long_traj)