From 806ab462d7e55d8e99596834cd8c6dac838c27e3 Mon Sep 17 00:00:00 2001
From: nhkcmany <nyhuis@tfd.uni-hannover.de>
Date: Wed, 23 Nov 2022 12:24:27 +0100
Subject: [PATCH] save figures option
---
ntrfc/cascade_case/case.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/ntrfc/cascade_case/case.py b/ntrfc/cascade_case/case.py
index 4fa6e8b6..bcfbd24c 100644
--- a/ntrfc/cascade_case/case.py
+++ b/ntrfc/cascade_case/case.py
@@ -20,7 +20,7 @@ class proberegistry:
new_probeset = probeset(variablename,groupname, time, values, timestamps)
self.probesets.append(new_probeset)
- def plot_all(self,id = "all"):
+ def plot_all(self,id = "all",save=None,dpi=300):
"""
:param id: str (all), int (id)
"""
@@ -55,9 +55,11 @@ class proberegistry:
plt.grid()
plt.legend()
plt.show()
+ if save:
+ plt.savefig(save,dpi=dpi)
plt.close()
- def plot_val(self,val):
+ def plot_val(self,val,save=None,dpi=300):
plt.figure(figsize=(32, 24),dpi=200)
for probeset in [i for i in self.probesets if i.variablename==val]:
nprobes = probeset.arrays.shape[1]
@@ -79,6 +81,8 @@ class proberegistry:
plt.grid()
plt.legend()
plt.show()
+ if save:
+ plt.savefig(save,dpi=dpi)
plt.close()
@dataclass
--
GitLab