From 4c73131c4ce1186b3388959feecf4164131adb06 Mon Sep 17 00:00:00 2001
From: Brian Guarraci <brian.guarraci@gmail.com>
Date: Fri, 5 Jan 2024 13:36:59 -0700
Subject: [PATCH] simplify the groupby

---
 klongpy/monads.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/klongpy/monads.py b/klongpy/monads.py
index 644799f..ce4cca4 100644
--- a/klongpy/monads.py
+++ b/klongpy/monads.py
@@ -198,8 +198,7 @@ def eval_monad_groupby(a):
     if len(q) == 0:
         return q
     a = q.argsort()
-    u = np.unique(q[a], return_index=True)
-    r = np.split(a, u[1][1:])
+    r = np.split(a, np.where(q[a][1:] != q[a][:-1])[0] + 1)
     return np.asarray(r, dtype=object)