Merge pull request #6147 from bburan-galenea/bburan/groupby-series-unique

ENH: Add unique and nunique to GroupBy whitelist

Author: jreback

pandas/core/groupby.py

@@ -75,7 +75,7 @@
 
 _series_apply_whitelist = \
     (_common_apply_whitelist - set(['boxplot'])) | \
-    frozenset(['dtype', 'value_counts'])
+    frozenset(['dtype', 'value_counts', 'unique', 'nunique'])
 
 _dataframe_apply_whitelist = \
     _common_apply_whitelist | frozenset(['dtypes', 'corrwith'])

pandas/tests/test_groupby.py

@@ -3335,6 +3335,7 @@ def test_groupby_whitelist(self):
             'corr', 'cov',
             'value_counts',
             'diff',
+            'unique', 'nunique',
         ])
 
         for obj, whitelist in zip((df, s),