Towards #13, K factor axis are now coloured

pymule/plot.py

@@ -158,10 +158,22 @@ def kplot(sigma, labelx='x_e', labelsigma=None,
     kwargs = {
         "labelx": labelx,
         "labupleft": labelsigma,
-        "labdownleft": labelknlo,
-        "labdownright": labelknnlo,
         "downalign": [1,1]
     }
+    if type(labelknlo) == dict:
+        kwargs["labdownleft"] = labelknlo
+    else:
+        kwargs["labdownleft"] = {
+            'ylabel': labelknlo,
+            'color': orderscheme['nlo']
+        }
+    if type(labelknlo) == dict:
+        kwargs["labdownright"] = labelknnlo
+    else:
+        kwargs["labdownright"] = {
+            'ylabel': labelknnlo,
+            'color': orderscheme['nnlo']
+        }
 
     if type(sigma) != dict:
         raise ValueError("kplot takes dicts!")
@@ -204,6 +216,10 @@ def kplot(sigma, labelx='x_e', labelsigma=None,
     ]
 
     fig, axs = twopanel(**kwargs)
+    if 1 in showk:
+        axs[1].tick_params(labelcolor=orderscheme['nlo'], axis='y')
+    if 2 in showk and 'nnlo' in sigma:
+        axs[2].tick_params(labelcolor=orderscheme['nnlo'], axis='y')
 
     axs[1].axhline(1, color='black', linewidth=1, zorder=1)