diff --git a/persim/visuals.py b/persim/visuals.py index 6d4c940c..56ef935f 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