Supplies tentative implementation for cumsum

hablapps · Nov 8, 2023 · 0fa1d5c · 0fa1d5c
1 parent 7de2b7c
commit 0fa1d5c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/pykx/pandas_api/pandas_meta.py b/src/pykx/pandas_api/pandas_meta.py
@@ -219,6 +219,17 @@ def any(self, axis=0, bool_only=False, skipna=True):
         res, cols = preparse_computations(self, axis, skipna, bool_only=bool_only)
         return (q('{"b"$x}', [any(x) for x in res]), cols)
 
+    @convert_result
+    def cumsum(self, axis=0, skipna=True):
+        res, cols = preparse_computations(self, axis, skipna)
+        return (q(
+            '{[row]' 
+            '{$[11h=type x;' 
+            '{[x1; y1] `$string[x1], string y1} scan x;' 
+            'sums x]} each row}', 
+            res
+        ), cols)
+
     @convert_result
     def max(self, axis=0, skipna=True, numeric_only=False):
         res, cols = preparse_computations(self, axis, skipna, numeric_only)