Improve overlap matrix plot. [ref #193]

michellab · Apr 27, 2021 · 383049c · 383049c
1 parent fadffe7
commit 383049c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/python/BioSimSpace/Notebook/_plot.py b/python/BioSimSpace/Notebook/_plot.py
@@ -542,8 +542,14 @@ def plotOverlapMatrix(overlap):
             text = ax.text(y, x, f"{overlap[x][y]:.2f}", ha="center", color="k")
 
     # Set the axis labels.
-    _plt.xlabel(r"$\lambda$")
-    _plt.ylabel(r"$\lambda$")
+    _plt.xlabel(r"$\lambda$ window")
+    _plt.ylabel(r"$\lambda$ window")
+
+    ticks = [x for x in range(0, num_rows)]
+
+    # Set ticks every lambda window.
+    _plt.xticks(ticks)
+    _plt.yticks(ticks)
 
     # Create a tight layout to trim whitespace.
     fig.tight_layout()