Finished a good part of analysis and figure for Thresh-LP invariance (WIP).

python/plot_functions.py

@@ -69,14 +69,22 @@ def ylimits(signal, ax=None, minval=None, maxval=None, pad=0.05):
         return ax.set_ylim(limits)
     return limits
 
-def xlabel(ax, label, y=-0.1, fontsize=20, **kwargs):
+def xlabel(ax, label, x=None, y=-0.1, fontsize=20, transform=None, **kwargs):
     ax.set_xlabel(label, fontsize=fontsize, **kwargs)
-    ax.xaxis.set_label_coords(0.5, y)
+    if x is None:
+        x = 0.5  
+        if transform is not None:
+            x = (ax.transAxes + transform.inverted()).transform((x, 0))[0]
+    ax.xaxis.set_label_coords(x, y, transform=transform)
     return None
 
-def ylabel(ax, label, x=-0.2, fontsize=20, **kwargs):
+def ylabel(ax, label, x=-0.2, y=None, fontsize=20, transform=None, **kwargs):
     ax.set_ylabel(label, fontsize=fontsize, **kwargs)
-    ax.yaxis.set_label_coords(x, 0.5)
+    if y is None:
+        y = 0.5
+        if transform is not None:
+            y = (ax.transAxes + transform.inverted()).transform((0, y))[1]
+    ax.yaxis.set_label_coords(x, y, transform=transform)
     return None
 
 def super_xlabel(label, fig, high_ax, low_ax, y=0.005, **kwargs):
@@ -136,3 +144,13 @@ def reorder_traces(handles, signal, zlow=2, zhigh=2.5):
         handles[ind].set_zorder(z)
     return None
 
+def strip_zeros(num, right_digits=5):
+    if isinstance(num, int):
+        return num
+    num = f'{num:.{right_digits}f}'
+    left, right = num.split('.')
+    right = right.rstrip('0')
+    if right:
+        return f'{left}.{right}'
+    return left
+