pandas-plots 0.12.17__tar.gz → 0.12.18__tar.gz

{pandas_plots-0.12.17/src/pandas_plots.egg-info → pandas_plots-0.12.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pandas-plots
-Version: 0.12.17
+Version: 0.12.18
 Summary: A collection of helper for table handling and visualization
 Home-page: https://github.com/smeisegeier/pandas-plots
 Author: smeisegeier

{pandas_plots-0.12.17 → pandas_plots-0.12.18}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = pandas-plots
-version = 0.12.17
+version = 0.12.18
 author = smeisegeier
 author_email = dexterDSDo@googlemail.com
 description = A collection of helper for table handling and visualization

{pandas_plots-0.12.17 → pandas_plots-0.12.18}/src/pandas_plots/hlp.py

@@ -471,7 +471,7 @@ pd.DataFrame.add_bitmask_label = add_bitmask_label
 ddb.DuckDBPyRelation.add_bitmask_label = add_bitmask_label
 
 
-def find_cols(all_cols: list[str], stubs=list[str]):
+def find_cols(all_cols: list[str], stubs: list[str] = None):
     """
     Find all columns in a list of columns that contain any of the given stubs.
 
@@ -487,10 +487,38 @@ def find_cols(all_cols: list[str], stubs=list[str]):
     list[str]
         List of columns that contain any of the given stubs.
     """
-    if all_cols is None or not stubs:
+    if all_cols is None or stubs is None:
         return "❌ empty lists"
     return [col for col in all_cols if any(match in col for match in stubs)]
 
+def find_cols(all_cols: list[str], stubs: list[str] = None) -> list[str]:
+    """
+    Find all columns in a list of columns that contain any of the given stubs,
+    preserving the order of stubs in the output.
+
+    Parameters
+    ----------
+    all_cols : list[str]
+        List of columns to search in.
+    stubs : list[str], optional
+        List of strings to search for in column names.
+
+    Returns
+    -------
+    list[str]
+        List of columns that contain any of the given stubs, ordered by stubs.
+    """
+    if all_cols is None or not stubs:
+        print("❌ empty lists")
+        return []
+    
+    result = []
+    for stub in stubs:
+        result.extend([col for col in all_cols if stub in col])
+    
+    return result
+
+
 # * extend objects to enable chaining
 pd.DataFrame.find_cols = find_cols
 

{pandas_plots-0.12.17 → pandas_plots-0.12.18/src/pandas_plots.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pandas-plots
-Version: 0.12.17
+Version: 0.12.18
 Summary: A collection of helper for table handling and visualization
 Home-page: https://github.com/smeisegeier/pandas-plots
 Author: smeisegeier