ENH: add option display.with_mpl_style GH3075

Author: y-p

RELEASE.rst

@@ -112,6 +112,9 @@ pandas 0.11.0
 
         df = DataFrame(dict(A = ts))
         df['2001']
+  - added option `display.mpl_style` providing a sleeker visual style
+    for plots. Based on https://gist.github.com/huyng/816622 (GH3075_).
+
 
 **API Changes**
 
@@ -239,9 +242,11 @@ pandas 0.11.0
 .. _GH2807: https://github.com/pydata/pandas/issues/2807
 .. _GH2918: https://github.com/pydata/pandas/issues/2918
 .. _GH3011: https://github.com/pydata/pandas/issues/3011
+.. _GH2745: https://github.com/pydata/pandas/issues/2745
+.. _GH622: https://github.com/pydata/pandas/issues/622
+.. _GH797: https://github.com/pydata/pandas/issues/797
 .. _GH2681: https://github.com/pydata/pandas/issues/2681
 .. _GH2719: https://github.com/pydata/pandas/issues/2719
-.. _GH2745: https://github.com/pydata/pandas/issues/2745
 .. _GH2746: https://github.com/pydata/pandas/issues/2746
 .. _GH2747: https://github.com/pydata/pandas/issues/2747
 .. _GH2751: https://github.com/pydata/pandas/issues/2751
@@ -282,7 +287,10 @@ pandas 0.11.0
 .. _GH3076: https://github.com/pydata/pandas/issues/3076
 .. _GH3063: https://github.com/pydata/pandas/issues/3063
 .. _GH3059: https://github.com/pydata/pandas/issues/3059
+.. _GH2993: https://github.com/pydata/pandas/issues/2993
 .. _GH3115: https://github.com/pydata/pandas/issues/3115
+.. _GH3070: https://github.com/pydata/pandas/issues/3070
+.. _GH3075: https://github.com/pydata/pandas/issues/3075
 .. _GH3130: https://github.com/pydata/pandas/issues/3130
 
 pandas 0.10.1

doc/source/v0.11.0.txt

@@ -301,6 +301,8 @@ Enhancements
   - DataFrame.from_records now accepts not only dicts but any instance of
     the collections.Mapping ABC.
 
+  - added option `display.with_wmp_style` providing a sleeker visual style
+    for plots. Based on https://gist.github.com/huyng/816622 (GH3075_).
 
 See the `full release notes
 <https://github.com/pydata/pandas/blob/master/RELEASE.rst>`__ or issue tracker
@@ -325,3 +327,4 @@ on GitHub for a complete list.
 .. _GH3076: https://github.com/pydata/pandas/issues/3076
 .. _GH3059: https://github.com/pydata/pandas/issues/3059
 .. _GH3070: https://github.com/pydata/pandas/issues/3070
+.. _GH3075: https://github.com/pydata/pandas/issues/3075

pandas/core/config_init.py

@@ -147,6 +147,38 @@
     perform a null check when repr'ing.
 """
 
+pc_mpl_style_doc = """
+: bool
+
+    Setting this to 'default' will modify the rcParams used by matplotlib
+    to give plots a more pleasing visual style by default.
+    Setting this to None/False restores the values to their initial value.
+"""
+
+style_backup = dict()
+def mpl_style_cb(key):
+    import sys
+    from pandas.tools.plotting import mpl_stylesheet
+    global style_backup
+
+    val = cf.get_option(key)
+
+    if 'matplotlib' not in sys.modules.keys():
+        if not(key): # starting up, we get reset to None
+            return val
+        raise Exception("matplotlib has not been imported. aborting")
+
+    import matplotlib.pyplot as plt
+
+
+    if val == 'default':
+        style_backup = dict([(k,plt.rcParams[k]) for k in mpl_stylesheet])
+        plt.rcParams.update(mpl_stylesheet)
+    elif not val:
+        if style_backup:
+            plt.rcParams.update(style_backup)
+
+    return val
 
 with cf.config_prefix('display'):
     cf.register_option('precision', 7, pc_precision_doc, validator=is_int)
@@ -177,6 +209,9 @@
     cf.register_option('line_width', 80, pc_line_width_doc)
     cf.register_option('chop_threshold', None, pc_chop_threshold_doc)
     cf.register_option('max_seq_items', None, pc_max_seq_items)
+    cf.register_option('mpl_style', None, pc_mpl_style_doc,
+                       validator=is_one_of_factory([None, False, 'default']),
+                       cb=mpl_style_cb)
 
 tc_sim_interactive_doc = """
 : boolean

pandas/tests/test_graphics.py

@@ -8,6 +8,7 @@
 from pandas import Series, DataFrame, MultiIndex, PeriodIndex, date_range
 import pandas.util.testing as tm
 from pandas.util.testing import ensure_clean
+from pandas.core.config import set_option,get_option,config_prefix
 
 import numpy as np
 
@@ -693,6 +694,21 @@ def test_grouped_hist(self):
         for ax in axes.ravel():
             self.assert_(len(ax.patches) > 0)
 
+    def test_option_mpl_style(self):
+        # just a sanity check
+        try:
+            import matplotlib
+        except:
+            raise nose.SkipTest
+
+        set_option('display.mpl_style', 'default')
+        set_option('display.mpl_style', None)
+        set_option('display.mpl_style', False)
+        try:
+            set_option('display.mpl_style', 'default2')
+        except ValueError:
+            pass
+
 def _check_plot_works(f, *args, **kwargs):
     import matplotlib.pyplot as plt
 

pandas/tools/plotting.py

@@ -23,6 +23,71 @@
 except ImportError:
     pass
 
+# Extracted from https://gist.github.com/huyng/816622
+# this is the rcParams set when setting display.with_mpl_style
+# to True.
+mpl_stylesheet = {
+    'axes.axisbelow': True,
+     'axes.color_cycle': ['#348ABD',
+      '#7A68A6',
+      '#A60628',
+      '#467821',
+      '#CF4457',
+      '#188487',
+      '#E24A33'],
+     'axes.edgecolor': '#bcbcbc',
+     'axes.facecolor': '#eeeeee',
+     'axes.grid': True,
+     'axes.labelcolor': '#555555',
+     'axes.labelsize': 'large',
+     'axes.linewidth': 1.0,
+     'axes.titlesize': 'x-large',
+     'figure.edgecolor': 'white',
+     'figure.facecolor': 'white',
+     'figure.figsize': (6.0, 4.0),
+     'figure.subplot.hspace': 0.5,
+     'font.family': 'monospace',
+     'font.monospace': ['Andale Mono',
+      'Nimbus Mono L',
+      'Courier New',
+      'Courier',
+      'Fixed',
+      'Terminal',
+      'monospace'],
+     'font.size': 10,
+     'interactive': True,
+     'keymap.all_axes': ['a'],
+     'keymap.back': ['left', 'c', 'backspace'],
+     'keymap.forward': ['right', 'v'],
+     'keymap.fullscreen': ['f'],
+     'keymap.grid': ['g'],
+     'keymap.home': ['h', 'r', 'home'],
+     'keymap.pan': ['p'],
+     'keymap.save': ['s'],
+     'keymap.xscale': ['L', 'k'],
+     'keymap.yscale': ['l'],
+     'keymap.zoom': ['o'],
+     'legend.fancybox': True,
+     'lines.antialiased': True,
+     'lines.linewidth': 1.0,
+     'patch.antialiased': True,
+     'patch.edgecolor': '#EEEEEE',
+     'patch.facecolor': '#348ABD',
+     'patch.linewidth': 0.5,
+     'toolbar': 'toolbar2',
+     'xtick.color': '#555555',
+     'xtick.direction': 'in',
+     'xtick.major.pad': 6.0,
+     'xtick.major.size': 0.0,
+     'xtick.minor.pad': 6.0,
+     'xtick.minor.size': 0.0,
+     'ytick.color': '#555555',
+     'ytick.direction': 'in',
+     'ytick.major.pad': 6.0,
+     'ytick.major.size': 0.0,
+     'ytick.minor.pad': 6.0,
+     'ytick.minor.size': 0.0
+}
 
 def _get_standard_kind(kind):
     return {'density': 'kde'}.get(kind, kind)