Add save option

Inst_eddies/Analysis/distribution_function.py

@@ -136,7 +136,15 @@ if __name__ == "__main__":
     parser.add_argument(
         "dir", help="directory containing collection of shapefiles"
     )
+    parser.add_argument("--save", action="store_true")
     args = parser.parse_args()
     d = read(args.dir)
     fig_list = plot_all([d])
-    plt.show()
+
+    if args.save:
+        for i, fig in enumerate(fig_list):
+            fig.savefig(f"figure_{i}.png")
+
+        print(f"Created figure_[0-{len(fig_list)-1}].png")
+    else:
+        plt.show()