Merge pull request #5787 from jorisvandenbossche/groupby-idxmax

Add idxmax/idxmin to groupby dispatch whitelist (#5786)

Author: jreback

pandas/core/groupby.py

@@ -65,6 +65,7 @@
     'mad',
     'any', 'all',
     'irow', 'take',
+    'idxmax', 'idxmin',
     'shift', 'tshift',
     'ffill', 'bfill',
     'pct_change', 'skew',

pandas/tests/test_groupby.py

@@ -3264,6 +3264,7 @@ def test_groupby_whitelist(self):
             'mad',
             'any', 'all',
             'irow', 'take',
+            'idxmax', 'idxmin',
             'shift', 'tshift',
             'ffill', 'bfill',
             'pct_change', 'skew',
@@ -3284,6 +3285,7 @@ def test_groupby_whitelist(self):
             'mad',
             'any', 'all',
             'irow', 'take',
+            'idxmax', 'idxmin',
             'shift', 'tshift',
             'ffill', 'bfill',
             'pct_change', 'skew',
@@ -3413,7 +3415,7 @@ def test_tab_completion(self):
             'resample', 'cummin', 'fillna', 'cumsum', 'cumcount',
             'all', 'shift', 'skew', 'bfill', 'irow', 'ffill',
             'take', 'tshift', 'pct_change', 'any', 'mad', 'corr', 'corrwith',
-            'cov', 'dtypes', 'diff',
+            'cov', 'dtypes', 'diff', 'idxmax', 'idxmin'
         ])
         self.assertEqual(results, expected)