add ungroup

thejaminator · Jan 16, 2024 · 2079217 · 2079217
1 parent 98a3e81
commit 2079217
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [tool]
 [tool.poetry]
 name = "slist"
-version = "0.3.1"
+version = "0.3.2"
 homepage = "https://github.com/thejaminator/slist"
 description = "A typesafe list with more method chaining!"
 authors = ["James Chua <chuajamessh@gmail.com>"]

diff --git a/slist/__init__.py b/slist/__init__.py
@@ -206,6 +206,11 @@ def __init__(self, name: str, age: int):
             else:
                 d[k] = Slist([elem])
         return Slist(Group(key=key, values=value) for key, value in d.items())
+
+
+    def ungroup(self: Slist[Group[A, B]]) -> Slist[B]:
+        """Ungroups the list of groups"""
+        return self.map_2(lambda _, values: values)
 
     def map_on_group_values(self: Slist[Group[B, Slist[C]]], func: Callable[[Slist[C]], D]) -> Slist[Group[B, D]]:
         # Apply a function on the group's vsalues