diff --git a/Trajectories/plot_traj.py b/Trajectories/plot_traj.py index 2b48d0eef0603e760ce2e65b001f8e7e52d45698..8f7bacf3cb73e324c144918bf0ecd741bb074a01 100755 --- a/Trajectories/plot_traj.py +++ b/Trajectories/plot_traj.py @@ -14,6 +14,8 @@ parser.add_argument("expanded_traj", help = "JSon file") parser.add_argument("e_overestim", help = "text file") parser.add_argument("SHPC", help = "directory") parser.add_argument("orientation", choices = ["Anticyclones", "Cyclones"]) +parser.add_argument("--save", metavar = "format", + help = "Save file to specified format") args = parser.parse_args() with open(args.expanded_traj) as f: @@ -51,4 +53,9 @@ for traj in trajectories: ax.coastlines() ax.gridlines(draw_labels = True) -plt.show() + +if args.save: + plt.savefig(f"plot_traj.{args.save}") + print(f'Created "plot_traj.{args.save}".') +else: + plt.show()