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

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.0.dist-info}/METADATA

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

qutepandas-1.1.0.dist-info/RECORD

@@ -0,0 +1,35 @@
+qutePandas/__init__.py,sha256=BO7UwZvxdpkHzCeXWasLf6mSVx-wOpaBNyXCDlEb_80,2012
+qutePandas/utils.py,sha256=X0GODOBxuAFOtjjFb6IGlzjDhhrkwH_6ErIMypiFhUM,1026
+qutePandas/apply/__init__.py,sha256=tdjeqTQI6jH530lTmpd9dkUPese6B22uI2P4ATKoGzc,92
+qutePandas/apply/apply.py,sha256=OqeNwgoD1G5Y6PK7G_2CX7fswq4_Fl0YDDaqWI7O-a4,2275
+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=NhYlNpHIlFMMd9ws0dKvw1njxUOmuf6VHLo8oQSUpEA,1057
+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=bwdm-MVIpoXRo0RblJ_crZVCJlUZgQLN4PmcYu6mdzk,3269
+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=g-fG0pxYHn7_qrBwDQ3MRsTi-U_VEtrN4_z5kDNmZaM,2272
+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.0.dist-info/METADATA,sha256=-8cJlJSnucXHNZMShlp5osW2NzdJvg2jcntF-qVsbXE,2540
+qutepandas-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qutepandas-1.1.0.dist-info/top_level.txt,sha256=O_6KkoxW8KjgZX5o7sYwYVb8XPtNBA7PRSTE08qda1k,11
+qutepandas-1.1.0.dist-info/RECORD,,

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,,