add sym option for logx logy loglog and tests

Author: charlesdong1991

pandas/plotting/_core.py

@@ -82,9 +82,9 @@ def _kind(self):
     _default_rot = 0
     orientation = None
     _pop_attributes = ['label', 'style', 'logy', 'logx', 'loglog',
-                       'mark_right', 'stacked']
+                       'mark_right', 'stacked', 'sym']
     _attr_defaults = {'logy': False, 'logx': False, 'loglog': False,
-                      'mark_right': True, 'stacked': False}
+                      'mark_right': True, 'stacked': False, 'sym': False}
 
     def __init__(self, data, kind=None, by=None, subplots=False, sharex=None,
                  sharey=False, use_index=True,
@@ -309,9 +309,15 @@ def _setup_subplots(self):
         axes = _flatten(axes)
 
         if self.logx or self.loglog:
-            [a.set_xscale('log') for a in axes]
+            if self.sym:
+                [a.set_xscale('symlog') for a in axes]
+            else:
+                [a.set_xscale('log') for a in axes]
         if self.logy or self.loglog:
-            [a.set_yscale('log') for a in axes]
+            if self.sym:
+                [a.set_yscale('symlog') for a in axes]
+            else:
+                [a.set_yscale('log') for a in axes]
 
         self.fig = fig
         self.axes = axes

pandas/tests/plotting/test_frame.py

@@ -233,12 +233,18 @@ def test_logscales(self):
         df = DataFrame({'a': np.arange(100)}, index=np.arange(100))
         ax = df.plot(logy=True)
         self._check_ax_scales(ax, yaxis='log')
+        ax = df.plot(logy=True, sym=True)
+        self._check_ax_scales(ax, yaxis='symlog')
 
         ax = df.plot(logx=True)
         self._check_ax_scales(ax, xaxis='log')
+        ax = df.plot(logx=True, sym=True)
+        self._check_ax_scales(ax, xaxis='symlog')
 
         ax = df.plot(loglog=True)
         self._check_ax_scales(ax, xaxis='log', yaxis='log')
+        ax = df.plot(loglog=True, sym=True)
+        self._check_ax_scales(ax, xaxis='symlog', yaxis='symlog')
 
     @pytest.mark.slow
     def test_xcompat(self):