ENH: KdePlot with DataFrame #1342. TST: frame kde and kde with logy #1341

Author: Chang She

pandas/tests/test_graphics.py

@@ -56,6 +56,11 @@ def test_hist(self):
         _check_plot_works(self.ts.hist)
         _check_plot_works(self.ts.hist, grid=False)
 
+    @slow
+    def test_kde(self):
+        _check_plot_works(self.ts.plot, kind='kde')
+        ax = self.ts.plot(kind='kde', logy=True)
+        self.assert_(ax.get_yscale() == 'log')
 
 class TestDataFramePlots(unittest.TestCase):
 
@@ -162,6 +167,15 @@ def test_boxplot(self):
 
         _check_plot_works(lambda x: plotting.boxplot(x), df['one'])
 
+    @slow
+    def test_kde(self):
+        df = DataFrame(np.random.randn(100, 4))
+        _check_plot_works(df.plot, kind='kde')
+        _check_plot_works(df.plot, kind='kde', subplots=True)
+        axes = df.plot(kind='kde', logy=True, subplots=True)
+        for ax in axes:
+            self.assert_(ax.get_yscale() == 'log')
+
     @slow
     def test_hist(self):
         df = DataFrame(np.random.randn(100, 4))

pandas/tools/plotting.py

@@ -678,6 +678,8 @@ def plot_frame(frame=None, subplots=False, sharex=True, sharey=False,
         klass = LinePlot
     elif kind in ('bar', 'barh'):
         klass = BarPlot
+    elif kind == 'kde':
+        klass = KdePlot
     else:
         raise ValueError('Invalid chart type given %s' % kind)