[plotstyle] new sketch_style(), plain_style() and plot_style()

plotstyle.py

@@ -21,6 +21,9 @@ colors = {
     'blue': '#0000CC'
 }
 
+# factor for scaling widths of bars in a bar plot:
+bar_fac = 1.0
+
 
 def cm_size(*args):
     """ Convert dimensions from cm to inch.
@@ -180,7 +183,7 @@ def __axis_label(label, unit=None):
         return '%s [%s]' % (label, unit)
 
 
-def __set_xlabel(ax, label, unit=None, **kwargs):
+def set_xlabel(ax, label, unit=None, **kwargs):
     """ Format the xlabel from a label and an unit.
 
     Uses the __axis_label() function to format the axis label.
@@ -197,7 +200,7 @@ def __set_xlabel(ax, label, unit=None, **kwargs):
     ax.set_xlabel_orig(__axis_label(label, unit), **kwargs)
 
         
-def __set_ylabel(ax, label, unit=None, **kwargs):
+def set_ylabel(ax, label, unit=None, **kwargs):
     """ Format the ylabel from a label and an unit.
 
     Uses the __axis_label() function to format the axis label.
@@ -214,7 +217,7 @@ def __set_ylabel(ax, label, unit=None, **kwargs):
     ax.set_ylabel_orig(__axis_label(label, unit), **kwargs)
 
         
-def __set_zlabel(ax, label, unit=None, **kwargs):
+def set_zlabel(ax, label, unit=None, **kwargs):
     """ Format the zlabel from a label and an unit.
 
     Uses the __axis_label() function to format the axis label.
@@ -231,47 +234,75 @@ def __set_zlabel(ax, label, unit=None, **kwargs):
     ax.set_zlabel_orig(__axis_label(label, unit), **kwargs)
 
 
-# overwrite axes constructor:
+def common_format():
-mpl.axes.Subplot.__init__orig = mpl.axes.Subplot.__init__
+    """ Set some rc parameter.
-mpl.axes.Subplot.__init__ = __axes__init__
+    """
-mpl.axes.Axes.show_spines = show_spines
+    mpl.rcParams['figure.figsize'] = cm_size(figure_width, figure_height)
-
+    mpl.rcParams['figure.subplot.left'] = 5.5*0.6*mpl.rcParams['font.size']/cm_size(figure_width)/ppi
-# overwrite axes set_[xyz]label() member functions:
+    mpl.rcParams['figure.subplot.right'] = 1.0 - 0.5*0.6*mpl.rcParams['font.size']/cm_size(figure_width)/ppi
-mpl.axes.Axes.set_xlabel_orig = mpl.axes.Axes.set_xlabel
+    mpl.rcParams['figure.subplot.bottom'] = 2.8*mpl.rcParams['font.size']/cm_size(figure_height)/ppi
-mpl.axes.Axes.set_xlabel = __set_xlabel
+    mpl.rcParams['figure.subplot.top'] = 1.0 - 0.5*mpl.rcParams['font.size']/cm_size(figure_height)/ppi
-mpl.axes.Axes.set_ylabel_orig = mpl.axes.Axes.set_ylabel
+    mpl.rcParams['figure.subplot.wspace'] = 0.4
-mpl.axes.Axes.set_ylabel = __set_ylabel
+    mpl.rcParams['figure.subplot.hspace'] = 0.6
-Axes3D.set_zlabel_orig = Axes3D.set_zlabel
+    mpl.rcParams['figure.facecolor'] = 'white'
-Axes3D.set_zlabel = __set_zlabel
+    mpl.rcParams['xtick.direction'] = 'out'
-
+    mpl.rcParams['ytick.direction'] = 'out'
-# initialization:
+    mpl.rcParams['xtick.major.width'] = 1.25
-if xkcd_style:
+    mpl.rcParams['ytick.major.width'] = 1.25
-    plt.xkcd()
+    # overwrite axes constructor:
+    if not hasattr(mpl.axes.Subplot, '__init__orig'):
+        mpl.axes.Subplot.__init__orig = mpl.axes.Subplot.__init__
+        mpl.axes.Subplot.__init__ = __axes__init__
+    mpl.axes.Axes.show_spines = show_spines
+    # overwrite axes set_[xyz]label() member functions:
+    if not hasattr(mpl.axes.Axes, 'set_xlabel_orig'):
+        mpl.axes.Axes.set_xlabel_orig = mpl.axes.Axes.set_xlabel
+        mpl.axes.Axes.set_xlabel = set_xlabel
+    if not hasattr(mpl.axes.Axes, 'set_ylabel_orig'):
+        mpl.axes.Axes.set_ylabel_orig = mpl.axes.Axes.set_ylabel
+        mpl.axes.Axes.set_ylabel = set_ylabel
+    if not hasattr(Axes3D, 'set_zlabel_orig'):
+        Axes3D.set_zlabel_orig = Axes3D.set_zlabel
+        Axes3D.set_zlabel = set_zlabel
+
+    
+def sketch_style():
+    """ Activate xkcd style and adapt some rc parameter.
+    """
+    global bar_fac
     bar_fac = 0.9
+    plt.xkcd()
+    common_format()
+    mpl.rcParams['legend.fontsize'] = 'medium'
+    mpl.rcParams['xtick.labelsize'] = 'medium'
+    mpl.rcParams['ytick.labelsize'] = 'medium'
     mpl.rcParams['xtick.major.size'] = 6
     mpl.rcParams['ytick.major.size'] = 6
-else:
+
+
+def plain_style():
+    """ Deactivate xkcd style and adapt some rc parameter.
+    """
+    global bar_fac
     bar_fac = 1.0
+    plt.rcdefaults()
+    common_format()
     mpl.rcParams['font.family'] = 'sans-serif'
+    mpl.rcParams['legend.fontsize'] = 'x-small'
     mpl.rcParams['xtick.labelsize'] = 'small'
     mpl.rcParams['ytick.labelsize'] = 'small'
     mpl.rcParams['xtick.major.size'] = 2.5
     mpl.rcParams['ytick.major.size'] = 2.5
-    mpl.rcParams['legend.fontsize'] = 'x-small'
-mpl.rcParams['figure.figsize'] = cm_size(figure_width, figure_height)
-mpl.rcParams['figure.subplot.left'] = 5.5*0.6*mpl.rcParams['font.size']/cm_size(figure_width)/ppi
-mpl.rcParams['figure.subplot.right'] = 1.0 - 0.5*0.6*mpl.rcParams['font.size']/cm_size(figure_width)/ppi
-mpl.rcParams['figure.subplot.bottom'] = 2.8*mpl.rcParams['font.size']/cm_size(figure_height)/ppi
-mpl.rcParams['figure.subplot.top'] = 1.0 - 0.5*mpl.rcParams['font.size']/cm_size(figure_height)/ppi
-mpl.rcParams['figure.subplot.wspace'] = 0.4
-mpl.rcParams['figure.subplot.hspace'] = 0.6
-mpl.rcParams['figure.facecolor'] = 'white'
-#mpl.rcParams['axes.spines.left'] = True    # newer matplotlib only
-#mpl.rcParams['axes.spines.bottom'] = True
-#mpl.rcParams['axes.spines.top'] = False
-#mpl.rcParams['axes.spines.right'] = False
-mpl.rcParams['xtick.direction'] = 'out'
-mpl.rcParams['ytick.direction'] = 'out'
-mpl.rcParams['xtick.major.width'] = 1.25
-mpl.rcParams['ytick.major.width'] = 1.25
 
+
+def plot_style():
+    """ Set rc parameter in dependence on xkcd_style.
+    """
+    if xkcd_style:
+        sketch_style()
+    else:
+        plain_style()
+
+
+# automatic initialization:
+plot_style()

regression/lecture/derivative.py

@@ -2,6 +2,8 @@ import numpy as np
 import matplotlib.pyplot as plt
 from plotstyle import *
 
+plain_style()
+
 fig = plt.figure( figsize=(2.5,3.4) )
 ax = fig.add_subplot(1, 1, 1)
 

regression/lecture/gradient.py

@@ -2,6 +2,8 @@ import numpy as np
 import matplotlib.pyplot as plt
 from plotstyle import *
 
+plain_style()
+
 fig, ax = plt.subplots( figsize=(3.5,3.7) )
 fig.subplots_adjust(**adjust_fs(fig, 4.0, 1.0, 2.8, 0.5))