#13 and #17: xi-cut plots are now a column

pymule/xicut.py

@@ -54,21 +54,16 @@ def addkeyedsets(sets):
     }
 
 
-def mergefkswithplot(sets, split=False, scale=1., showfit=[True,True]):
+def mergefkswithplot(sets, scale=1., showfit=[True,True]):
     n = len(sets)
     psets = [addkeyedsets(i) for i in sets]
     ans = mergeset(addkeyedsets(psets).values())
 
-    if split:
-        fig1, ax1 = plt.subplots()
-        fig2, ax2 = plt.subplots()
-        fig = [fig1,fig2]
-    else:
-        fig, (ax1, ax2) = plt.subplots(1, 2)
+    fig, (ax1,ax2) = plt.subplots(
+        2, sharex=True, gridspec_kw={'hspace': 0, 'height_ratios': [3,2]}
+    )
 
-    ax1.set_xscale('log')
     ax2.set_xscale('log')
-    ax1.set_xlabel('$\\xi_c$')
     ax2.set_xlabel('$\\xi_c$')
     ax1.set_ylabel('$\\sigma^{(%d)}_i$' % (n-1))
     ax2.set_ylabel(
@@ -131,15 +126,20 @@ def mergefkswithplot(sets, split=False, scale=1., showfit=[True,True]):
         c='C0'
     )
 
+    # Matplotlib's offset placing routine has a bug. This sorta fixes it
+    ax2.yaxis._update_offset_text_position=lambda a,b : 0
+    ax2.yaxis.offsetText.set_transform(ax2.transAxes)
+    ax2.yaxis.offsetText.set_verticalalignment('top')
+    ax2.yaxis.offsetText.set_position((
+        0,
+        0.98
+    ))
+
     ax1.legend(
         ['$\\sigma^{(%d)}_{n}$' % (n-1)] +
         ['$\\sigma^{(%d)}_{n+%d}$' % (i, n-1) for i in range(1, n)] +
         ['$\\sigma^{(%d)}$' % (n-1)],
         loc='upper right'
     )
-    ax2.legend(
-        ['$\\sigma^{(%d)}$' % (n-1)],
-        loc='upper right'
-    )
 
     return fig, ans