From 493b834125b794c3d69be2391e6ef0a44f8c2033 Mon Sep 17 00:00:00 2001
From: Lionel GUEZ <guez@lmd.ipsl.fr>
Date: Fri, 11 Oct 2024 19:23:41 +0200
Subject: [PATCH] Add save option
---
Inst_eddies/Analysis/distribution_function.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/Inst_eddies/Analysis/distribution_function.py b/Inst_eddies/Analysis/distribution_function.py
index 400d4092..2f661173 100755
--- a/Inst_eddies/Analysis/distribution_function.py
+++ b/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()
--
GitLab