Add client support for SeriesGroupby unique, nsmallest, nlargest (mod…

…in-project#63) * Add client support for SeriesGroupby unique, nsmallest, nlargest Signed-off-by: Naren Krishna <[email protected]> --------- Signed-off-by: Naren Krishna <[email protected]>
mvashishtha · Feb 10, 2023 · a72e66a · a72e66a
1 parent 5383ff2
commit a72e66a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/modin/core/execution/client/query_compiler.py b/modin/core/execution/client/query_compiler.py
@@ -944,6 +944,7 @@ def forwarding_method(self, by, *args, **kwargs):
         "std",
         "sem",
         "rank",
+        "unique",
         "nunique",
         "median",
         "quantile",
@@ -954,6 +955,8 @@ def forwarding_method(self, by, *args, **kwargs):
         "var",
         "first",
         "last",
+        "nlargest",
+        "nsmallest",
     }
 )
 

diff --git a/modin/pandas/groupby.py b/modin/pandas/groupby.py
@@ -1307,6 +1307,31 @@ def _iter(self):
                 for k in (sorted(group_ids) if self._sort else group_ids)
             )
 
+    def unique(self):
+        return self._check_index(
+            self._wrap_aggregation(
+                type(self._query_compiler).groupby_unique,
+                numeric_only=False,
+            )
+        )
+
+    def nlargest(self, n=5, keep="first"):
+        return self._check_index(
+            self._wrap_aggregation(
+                type(self._query_compiler).groupby_nlargest,
+                agg_kwargs=dict(n=n, keep=keep),
+                numeric_only=True,
+            )
+        )
+
+    def nsmallest(self, n=5, keep="first"):
+        return self._check_index(
+            self._wrap_aggregation(
+                type(self._query_compiler).groupby_nsmallest,
+                agg_kwargs=dict(n=n, keep=keep),
+                numeric_only=True,
+            )
+        )
 
 if IsExperimental.get():
     from modin.experimental.cloud.meta_magic import make_wrapped_class