add .groupBy operations

gerlichlab · Oct 29, 2024 · 14cb6a4 · 14cb6a4
1 parent cae5099
commit 14cb6a4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/modules/pan/src/main/scala/collections.scala b/modules/pan/src/main/scala/collections.scala
@@ -104,6 +104,9 @@ object collections:
           */
         def contains(x: X): Boolean = (xs: Set[X]).contains(x)
 
+        /** Use the underlying collection's {@code .groupBy} operation */
+        def groupBy[K](f: X => K): Map[K, Set[X]] = (xs: Set[X]).groupBy(f)
+
         /** Convert safely to [[cats.data.NonEmptySet]]. */
         def toNes(using ord: Order[X]): NonEmptySet[X] =
           NonEmptySet.fromSetUnsafe(xs.toSortedSet)
@@ -120,6 +123,9 @@ object collections:
           */
         def contains(x: X): Boolean = (xs: List[X]).contains(x)
 
+        /** Use the underlying collection's {@code .groupBy} operation */
+        def groupBy[K](f: X => K): Map[K, List[X]] = (xs: List[X]).groupBy(f)
+
         /** Access the first element of the collection. */
         def head: X = (xs: List[X]).head