add "axis='xy'" option to grid() for independent control of each axis' grid.

kaufman · kaufman · commit b73236b59cbf · 2011-04-09T23:36:23.000-04:00
modified:   lib/matplotlib/axes.py
	modified:   lib/matplotlib/pyplot.py
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -2036,11 +2036,11 @@ def set_axisbelow(self, b):
         self._axisbelow = b
 
     @docstring.dedent_interpd
-    def grid(self, b=None, which='major', **kwargs):
+    def grid(self, b=None, which='major', axis='xy', **kwargs):
         """
         call signature::
 
-           grid(self, b=None, which='major', **kwargs)
+           grid(self, b=None, which='major', axis='xy', **kwargs)
 
         Set the axes grids on or off; *b* is a boolean.  (For MATLAB
         compatibility, *b* may also be a string, 'on' or 'off'.)
@@ -2052,6 +2052,9 @@ def grid(self, b=None, which='major', **kwargs):
         *which* can be 'major' (default), 'minor', or 'both' to control
         whether major tick grids, minor tick grids, or both are affected.
 
+        *axis* can be 'xy' (default), 'x', or 'y' to control which
+        set of gridlines are drawn.
+
         *kawrgs* are used to set the grid line properties, eg::
 
            ax.grid(color='r', linestyle='-', linewidth=2)
@@ -2064,8 +2067,11 @@ def grid(self, b=None, which='major', **kwargs):
         if len(kwargs):
             b = True
         b = _string_to_bool(b)
-        self.xaxis.grid(b, which=which, **kwargs)
-        self.yaxis.grid(b, which=which, **kwargs)
+
+        if axis == 'x' or  axis == 'xy':
+          self.xaxis.grid(b, which=which, **kwargs)
+        if axis == 'y' or  axis == 'xy':
+          self.yaxis.grid(b, which=which, **kwargs)
 
     def ticklabel_format(self, **kwargs):
         """
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -2644,8 +2644,8 @@ def cla():
 # This function was autogenerated by boilerplate.py.  Do not edit as
 # changes will be lost
 @docstring.copy_dedent(Axes.grid)
-def grid(b=None, which='major', **kwargs):
-    ret =  gca().grid(b, which, **kwargs)
+def grid(b=None, which='major', axis='xy', **kwargs):
+    ret =  gca().grid(b, which, axis, **kwargs)
     draw_if_interactive()
     return ret
 
