From 60c606edd4307780faa4304f09c0b3bb4a11c725 Mon Sep 17 00:00:00 2001 From: Tor Erik Larsen Date: Tue, 24 Sep 2019 15:59:49 +0200 Subject: [PATCH] add pyplot savefig() --- persim/visuals.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/persim/visuals.py b/persim/visuals.py index 6d4c940..56ef935 100644 --- a/persim/visuals.py +++ b/persim/visuals.py @@ -17,7 +17,10 @@ def plot_diagrams( lifetime=False, legend=True, show=False, - ax=None + ax=None, + save=False, + figures_directory=None, + filename=None ): """A helper function to plot persistence diagrams. @@ -175,6 +178,9 @@ def plot_diagrams( if show is True: plt.show() + if save is True: + plt.savefig(figures_directory + filename) + def bottleneck_matching(I1, I2, matchidx, D, labels=["dgm1", "dgm2"], ax=None): """ Visualize bottleneck matching between two diagrams