Merge pull request #50 from vonpost/master

Made default labels in plot_diagrams correspond to the number of diagrams

Author: sauln

RELEASE.txt

@@ -1,3 +1,6 @@
+0.2.1
+    - Allowed for more than 9 diagram labels in plot_persistence_diagrams.
+
 0.2.0
     - New full featured implementation of Persistence Images.
     - Legacy PersImage now deprecated.

persim/_version.py

@@ -1 +1 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"

persim/visuals.py

@@ -67,24 +67,14 @@ def plot_diagrams(
 
     xlabel, ylabel = "Birth", "Death"
 
-    if labels is None:
-        # Provide default labels for diagrams if using self.dgm_
-        labels = [
-            "$H_0$",
-            "$H_1$",
-            "$H_2$",
-            "$H_3$",
-            "$H_4$",
-            "$H_5$",
-            "$H_6$",
-            "$H_7$",
-            "$H_8$",
-        ]
-
     if not isinstance(diagrams, list):
         # Must have diagrams as a list for processing downstream
         diagrams = [diagrams]
 
+    if labels is None:
+        # Provide default labels for diagrams if using self.dgm_
+        labels = ["$H_{{{}}}$".format(i) for i , _ in enumerate(diagrams)]
+
     if plot_only:
         diagrams = [diagrams[i] for i in plot_only]
         labels = [labels[i] for i in plot_only]
@@ -281,4 +271,4 @@ def wasserstein_matching(I1, I2, matchidx, palette=None, labels=["dgm1", "dgm2"]
         else:
             plt.plot([I1[i, 0], I2[j, 0]], [I1[i, 1], I2[j, 1]], "g")
 
-    plot_diagrams([I1, I2], labels=labels, ax=ax)
+    plot_diagrams([I1, I2], labels=labels, ax=ax)