diff options
author | Vincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com> | 2022-10-19 09:17:11 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-19 09:17:11 +0200 |
commit | 6d19678c8ec66ddb3a7fa7327c0d1a1d5e5a3919 (patch) | |
tree | 9340d5543744f99065f2cfcdba163c30f6cf8101 /src | |
parent | f310138410bf9a90238cd9b75f79727cba0e3a95 (diff) | |
parent | e94e490554ce56040b998893977255a704d35e59 (diff) |
Merge pull request #706 from mglisse/bars
Fix badly centered barcode
Diffstat (limited to 'src')
-rw-r--r-- | src/python/gudhi/persistence_graphical_tools.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/python/gudhi/persistence_graphical_tools.py b/src/python/gudhi/persistence_graphical_tools.py index 21275cdd..e438aa66 100644 --- a/src/python/gudhi/persistence_graphical_tools.py +++ b/src/python/gudhi/persistence_graphical_tools.py @@ -194,19 +194,21 @@ def plot_persistence_barcode( y=[(death - birth) if death != float("inf") else (infinity - birth) for (dim,(birth,death)) in persistence] c=[colormap[dim] for (dim,(birth,death)) in persistence] - axes.barh(list(reversed(range(len(x)))), y, height=0.8, left=x, alpha=alpha, color=c, linewidth=0) + axes.barh(range(len(x)), y, left=x, alpha=alpha, color=c, linewidth=0) if legend: - dimensions = list(set(item[0] for item in persistence)) + dimensions = set(item[0] for item in persistence) axes.legend( handles=[mpatches.Patch(color=colormap[dim], label=str(dim)) for dim in dimensions], loc="lower right", ) axes.set_title("Persistence barcode", fontsize=fontsize) + axes.set_yticks([]) + axes.invert_yaxis() # Ends plot on infinity value and starts a little bit before min_birth if len(x) != 0: - axes.axis([axis_start, infinity, 0, len(x)]) + axes.set_xlim((axis_start, infinity)) return axes except ImportError as import_error: |