ENH: Groupby.plot enhancement

Author: sinhrks

pandas/core/groupby.py

@@ -3279,9 +3279,13 @@ def _apply_to_column_groupbys(self, func):
              in self._iterate_column_groupbys()),
             keys=self._selected_obj.columns, axis=1)
 
-from pandas.tools.plotting import boxplot_frame_groupby
+
+from pandas.tools.plotting import boxplot_frame_groupby, plot_grouped
 DataFrameGroupBy.boxplot = boxplot_frame_groupby
 
+SeriesGroupBy.plot = plot_grouped
+DataFrameGroupBy.plot = plot_grouped
+
 
 class PanelGroupBy(NDFrameGroupBy):
 

pandas/tests/test_groupby.py

@@ -4606,9 +4606,10 @@ def test_frame_groupby_plot_boxplot(self):
         df = DataFrame({'height': height, 'weight': weight, 'gender': gender})
         gb = df.groupby('gender')
 
-        res = gb.plot()
-        self.assertEqual(len(plt.get_fignums()), 2)
-        self.assertEqual(len(res), 2)
+        with assertRaisesRegexp(ValueError, "To plot DataFrameGroupBy, specify 'suplots=True'"):
+            res = gb.plot()
+            self.assertEqual(len(plt.get_fignums()), 2)
+            self.assertEqual(len(res), 2)
         tm.close()
 
         res = gb.boxplot()