qutePandas 1.0.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

qutePandas/transformation/cast.py

@@ -0,0 +1,71 @@
+import pykx as kx
+import pandas as pd
+from ..utils import _ensure_q_table, _handle_return
+
+
+def cast(df, col, dtype, return_type='q'):
+    """
+    Converts column to specified data type.
+
+    Parameters
+    ----------
+    df : pandas.DataFrame or pykx.Table
+        Input DataFrame.
+    col : str
+        Column name to cast.
+    dtype : str
+        Target data type ('i' for int, 'f' for float, 's' for symbol, etc.).
+    return_type : str, default 'q'
+        Desired return type ('p' or 'q').
+
+    Returns
+    -------
+    pandas.DataFrame or pykx.Table
+        DataFrame with column cast to new type.
+    """
+    try:
+        q_map = {
+            'int64': 'j', 'int32': 'i', 'int': 'i', 'long': 'j',
+            'float64': 'f', 'float32': 'e', 'float': 'f', 'real': 'e',
+            'object': 's', 'string': 'C', 'str': 'C',
+            'j': 'j', 'i': 'i', 'h': 'h', 'f': 'f', 'e': 'e', 's': 's', 'c': 'c'
+        }
+
+        q_type = q_map.get(dtype, dtype)
+        q_table = _ensure_q_table(df)
+
+        if len(q_type) != 1:
+            raise ValueError(f"Unsupported q cast type: {dtype}")
+
+        curr_type = kx.q(f'{{type x`{col}}}', q_table).py()
+        type_to_code = {
+            'j': 7, 'i': 6, 'h': 5, 'f': 9, 'e': 8, 's': 11, 'c': 10, 'b': 1
+        }
+        target_code = type_to_code.get(q_type.lower())
+
+        if target_code is not None and abs(curr_type) == target_code:
+            return _handle_return(q_table, return_type)
+
+        is_parsing = curr_type in (0, 10)
+        q_char = q_type.upper() if is_parsing else q_type.lower()
+
+        if q_char.lower() == 's':
+            result = kx.q(
+                f'{{update {col}:`$ string {col} from x}}',
+                q_table
+            )
+        elif q_char in ('i', 'j'):
+            result = kx.q(
+                f'{{update {col}:"{q_char}"$(({col}>=0)*floor {col} + ({col}<0)*ceiling {col}) from x}}',
+                q_table
+            )
+        else:
+            result = kx.q(
+                f'{{update {col}:"{q_char}"${col} from x}}',
+                q_table
+            )
+
+        return _handle_return(result, return_type)
+
+    except Exception as e:
+        raise RuntimeError(f"Failed to cast column {col} to type {dtype}: {e}")

qutePandas/transformation/drop_col.py

@@ -0,0 +1,53 @@
+import pykx as kx
+import pandas as pd
+from ..utils import _ensure_q_table, _handle_return, _validate_columns
+
+
+def drop_col(df, cols, return_type='q'):
+    """
+    Removes specified column(s) from the DataFrame.
+
+    Parameters
+    ----------
+    df : pandas.DataFrame or pykx.Table
+        Input DataFrame.
+    cols : list of str or str
+        Column names to drop.
+    return_type : str, default 'q'
+        Desired return type ('p' or 'q').
+
+    Returns
+    -------
+    pandas.DataFrame or pykx.Table
+        DataFrame with specified column(s) removed.
+    """
+    try:
+        q_table = _ensure_q_table(df)
+
+        if isinstance(cols, str):
+            cols = [cols]
+
+        if not isinstance(cols, (list, tuple)):
+            raise ValueError("cols must be a list of column names")
+
+        if not cols:
+            return _handle_return(q_table, return_type)
+
+        _validate_columns(q_table, cols)
+        
+        all_cols = kx.q("cols", q_table).py()
+        if len(cols) == len(all_cols):
+            if return_type == 'p':
+                count = kx.q('count', q_table).py()
+                return pd.DataFrame(index=range(count))
+            else:
+                result = kx.q('(0#`)!()')
+                return result
+        else:
+            q_cols = kx.SymbolVector(cols)
+            result = kx.q('{[t;c] ![t;();0b;(),c]}', q_table, q_cols)
+
+        return _handle_return(result, return_type)
+
+    except Exception as e:
+        raise RuntimeError(f"Failed to drop columns {cols}: {e}")

qutePandas/transformation/rename.py

@@ -0,0 +1,35 @@
+import pykx as kx
+import pandas as pd
+from ..utils import _ensure_q_table, _handle_return
+
+
+def rename(df, columns, return_type='q'):
+    """
+    Renames columns in the DataFrame.
+
+    Parameters
+    ----------
+    df : pandas.DataFrame or pykx.Table
+        Input DataFrame.
+    columns : dict
+        Dictionary mapping old column names to new column names.
+    return_type : str, default 'q'
+        Desired return type ('p' or 'q').
+
+    Returns
+    -------
+    pandas.DataFrame or pykx.Table
+        DataFrame with renamed columns.
+    """
+    try:
+        q_table = _ensure_q_table(df)
+        
+        cols = kx.q("cols", q_table).py()
+        new_cols = [columns.get(c, c) for c in cols]
+        new_cols_str = "`" + "`".join(new_cols)
+        
+        result = kx.q(f'{new_cols_str} xcol', q_table)
+        return _handle_return(result, return_type)
+    except Exception as e:
+        raise RuntimeError(f"Failed to rename columns: {e}")
+ 

{qutepandas-1.0.0.dist-info → qutepandas-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qutePandas
-Version: 1.0.0
+Version: 1.1.1
 Summary: A pandas-like library for kdb+/q using pykx
 Home-page: https://ishapatro.github.io/qutePandas/
 Author: Isha Patro

qutepandas-1.1.1.dist-info/RECORD

@@ -0,0 +1,35 @@
+qutePandas/__init__.py,sha256=QbVhBW6oE6B-f5BqIx_TNme6rk-zhSKb2iqe0RWAcFE,3143
+qutePandas/utils.py,sha256=X0GODOBxuAFOtjjFb6IGlzjDhhrkwH_6ErIMypiFhUM,1026
+qutePandas/apply/__init__.py,sha256=tdjeqTQI6jH530lTmpd9dkUPese6B22uI2P4ATKoGzc,92
+qutePandas/apply/apply.py,sha256=5Fa4I7_zkILk0TMJioWoAFh9kgrb-jVI4PMBcjIkWmQ,2465
+qutePandas/apply/apply_col.py,sha256=h6nAsAXLIC4WcMD-amW3bXkaNEe30SgKEBfUr3eKTng,1597
+qutePandas/cleaning/__init__.py,sha256=6tRzTPgCEumZ7c_TvsMit3pRxZhdzf1Pd4y23ozv_2k,204
+qutePandas/cleaning/dropna.py,sha256=SGIPg8vk4PB1jtnH8hDComuMz8omdraJhOPzKTRiUgI,739
+qutePandas/cleaning/dropna_col.py,sha256=hpR4BN-ZuYdz222aUlYKyrEA7QmYomAMqTK6ebVB0ro,831
+qutePandas/cleaning/fillna.py,sha256=9PDZCM1hyLtxFkVfQz7NJu1xrjDSW-nPQU47hysfDTI,1689
+qutePandas/cleaning/remove_duplicates.py,sha256=zf6TTbkuSh5S4ZCmgznPLl-erw-umVP7K0EqOyv9wUM,757
+qutePandas/core/__init__.py,sha256=o8WuiBhI5jBqKXTL80kBTNcxZo1aO45dQx2Z6UwWg50,311
+qutePandas/core/connection.py,sha256=CsBWI2jwIO1kNicA42AefmlRgVx5WqayCJPhHzJHhFc,4275
+qutePandas/core/dataframe.py,sha256=ehfz8lMIV_Yj1f9bMpY25TH3EMbR02Dvji9lhGg395E,3209
+qutePandas/core/display.py,sha256=ubSZM8yf0kARuiSXpUHJ0WmIt4syjkp-VuV2NAXbiGc,5058
+qutePandas/grouping/__init__.py,sha256=GOBRAWDuh3PnKHEVDzmEqoG_GIfU4G6Gtk0aeSG92lE,116
+qutePandas/grouping/groupby_avg.py,sha256=F8Gc6XdOKiWyNKOoDURgYeeIxIduzbds2mHCfxEtvw4,1129
+qutePandas/grouping/groupby_sum.py,sha256=9XZ2tITeCMRtdnpwH2CpoQh3Y3NDC-ufwiDg6EQUJOA,1128
+qutePandas/indexing/__init__.py,sha256=2MmNlO54FOdIaaPSzegum3QmeO-Ts8ya3xSCodQjTIo,45
+qutePandas/indexing/iloc.py,sha256=g0LYKZAqdVk6Xz--RUdb-VtECnzFRET28qh6KL5SRgE,3117
+qutePandas/indexing/loc.py,sha256=Xn_043CrKyJBTXhRUYWtU0ZF1yikIGxTI0gTJAYIEfA,1511
+qutePandas/introspection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+qutePandas/introspection/dtypes.py,sha256=h3qImsu3JBuELp2zeIfwyOlFTsC513U_qSRNuYzeXWw,705
+qutePandas/io/__init__.py,sha256=sX93I3DAXM_kT0g67Nlhj_tAlVhtQe9Tr4AbgbEle3c,92
+qutePandas/io/from_csv.py,sha256=QQ1lfyXGljvuJ4aL_RRaX7loRgfQ524XvWtBo0fTIHc,671
+qutePandas/io/to_csv.py,sha256=7VrLtEjIUT-gIBXCYhW5GnbiRR2HqAZK6OW8Gy0ZOLM,926
+qutePandas/joining/__init__.py,sha256=INM7LAOy5KqMIUrJdRHt7-no7ODiHsyx4F68hRbddYs,45
+qutePandas/joining/merge.py,sha256=q9-X7QC7omhobO27q2aUego557ewh5RxALzAP7Mgq2M,3854
+qutePandas/transformation/__init__.py,sha256=ECPmFXZq-uh5ajkgAVQbdc8flVzrcwkhEqwiOqqDeag,123
+qutePandas/transformation/cast.py,sha256=VD8D0gPs9GJX52bqu_LKt7BHU69Iv-scOq-E2Q_MGQU,2226
+qutePandas/transformation/drop_col.py,sha256=4hdGjitDbYSfxj-4jo73NmTa2qKed-NjGj0jGBfiANY,1509
+qutePandas/transformation/rename.py,sha256=MwWjbhaZCUdQgkObaMPIB8fUY240x_EdD41cwszCa4o,948
+qutepandas-1.1.1.dist-info/METADATA,sha256=F9226k9uz3MbdZhGJrzEveD7stG-JZPMsghZ4-6ySNE,2540
+qutepandas-1.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+qutepandas-1.1.1.dist-info/top_level.txt,sha256=O_6KkoxW8KjgZX5o7sYwYVb8XPtNBA7PRSTE08qda1k,11
+qutepandas-1.1.1.dist-info/RECORD,,

{qutepandas-1.0.0.dist-info → qutepandas-1.1.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

qutepandas-1.0.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-qutePandas/__init__.py,sha256=BO7UwZvxdpkHzCeXWasLf6mSVx-wOpaBNyXCDlEb_80,2012
-qutePandas/utils.py,sha256=X0GODOBxuAFOtjjFb6IGlzjDhhrkwH_6ErIMypiFhUM,1026
-qutepandas-1.0.0.dist-info/METADATA,sha256=x6JQlnuk6MwcDaLy7SkkWu1YcC4TAml13Z4PWRE5l5c,2540
-qutepandas-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qutepandas-1.0.0.dist-info/top_level.txt,sha256=O_6KkoxW8KjgZX5o7sYwYVb8XPtNBA7PRSTE08qda1k,11
-qutepandas-1.0.0.dist-info/RECORD,,