25: allowed for options to be passed to label in #17

pymule/plot.py

@@ -91,16 +91,25 @@ def twopanel(labelx,
     fig, axs = plt.subplots(
         2, sharex=True, gridspec_kw={'hspace': 0}
     )
-    axs[1].set_xlabel(labelx)
+    if type(labelx) == str:
+        axs[1].set_xlabel(labelx)
+    else:
+        axs[1].set_xlabel(**labelx)
 
-    axs[0].set_ylabel(labupleft)
+    if type(labupleft) == str:
+        axs[0].set_ylabel(labupleft)
+    else:
+        axs[0].set_ylabel(**labupleft)
     for i, c in zip(upleft, colupleft):
         errorband(i, ax=axs[0], col=c)
 
     if len(upright) > 0:
         ax2 = axs[0].twinx()
         if labupright is not None:
-            ax2.set_ylabel(labupright)
+            if type(labupright) == str:
+                ax2.set_ylabel(labupright)
+            else:
+                ax2.set_ylabel(**labupright)
         for i, c in zip(upright, colupright):
             errorband(i, ax=ax2, col=c)
         if len(upalign) == 2:
@@ -109,14 +118,20 @@ def twopanel(labelx,
                 ax2, upalign[1]
             )
 
-    axs[1].set_ylabel(labdownleft)
+    if type(labdownleft) == str:
+        axs[1].set_ylabel(labdownleft)
+    else:
+        axs[1].set_ylabel(**labdownleft)
     for i, c in zip(downleft, coldownleft):
         errorband(i, ax=axs[1], col=c)
 
     if len(downright) > 0:
         ax3 = axs[1].twinx()
         if labdownright is not None:
-            ax3.set_ylabel(labdownright)
+            if type(labdownright) == str:
+                ax3.set_ylabel(labdownright)
+            else:
+                ax3.set_ylabel(**labdownright)
         for i, c in zip(downright, coldownright):
             errorband(i, ax=ax3, col=c)
         if len(downalign) == 2:
@@ -136,8 +151,10 @@ def kplot(sigma, labelx='x_e', labelsigma=None,
         labelsigma = r"$\D\sigma\,/\,\D " + labelx + "$"
 
     kwargs = {
-        "labelx":"$%s$" % labelx,
+        "labelx": labelx,
         "labupleft": labelsigma,
+        "labdownleft": labelknlo,
+        "labdownright": labelknnlo,
         "downalign": [1,1]
     }