Values first approach

KxSystems · Dec 21, 2023 · 82d276d · 82d276d
1 parent d28e93a
commit 82d276d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/pykx/pandas_api/pandas_meta.py b/src/pykx/pandas_api/pandas_meta.py
@@ -257,6 +257,11 @@ def sum(self, axis=0, skipna=True, numeric_only=False, min_count=0):
             min_count
         ), cols)
 
+    @api_return
+    def values(self, skipna = True, numeric_only = False):
+        tab = self
+        return q('{[table] flip value flip table}', tab)
+
     def agg(self, func, axis=0, *args, **kwargs): # noqa: C901
         if 'KeyedTable' in str(type(self)):
             raise NotImplementedError("'agg' method not presently supported for KeyedTable")

diff --git a/tests/test_pandas_api.py b/tests/test_pandas_api.py
@@ -2029,3 +2029,11 @@ def test_keyed_loc_fixes(q):
         mkt[['k1', 'y']]
     with pytest.raises(KeyError):
         mkt['k1']
+
+def test_values(q):
+    col1 = q('10?100')
+    col2 = q('10?`a`b`c`d')
+    col3 = q('10?`x`y`z`1`2`3')
+    matrix = q('{[r1; r2; r3] flip enlist[r1; r2; r3]}', col1, col2, col3)
+    tab = q('{[price; sym; id] ([]p: price; s: sym; i: id)}', col1, col2, col3)
+    assert tab.values().py() == matrix.py()