pandas-plots 0.12.16__py3-none-any.whl → 0.12.18__py3-none-any.whl

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/pls.py

@@ -661,7 +661,6 @@ def plot_bars(
 
     # * layot caption if provided
     caption = _set_caption(caption)
-
     # ! plot
     _fig = px.bar(
         df.sort_values(
@@ -674,7 +673,7 @@ def plot_bars(
         orientation=orientation,
         # * retrieve the original columns from series
         title=title
-        or f"{caption}{_title_str_minval}{_title_str_top}[{df.columns[1]}] by [{col_index}]{_title_str_null}{_title_str_n}",
+        or f"{caption}{_title_str_minval}{_title_str_top}[{col_name}] by [{col_index}]{_title_str_null}{_title_str_n}",
         # * retrieve theme from env (intro.set_theme) or default
         template="plotly_dark" if os.getenv("THEME") == "dark" else "plotly",
         width=width,
@@ -732,8 +731,22 @@ def plot_bars(
 
     # * looks better on single bars
     _fig.update_traces(
-        textposition="outside" if not use_ci else "auto", error_y=dict(thickness=5)
+        error_y=dict(thickness=5)
     )
+    if use_ci:
+        _fig.update_traces(
+            textposition="inside",  # Put labels inside bars
+            insidetextanchor="start",  # Align labels at the bottom
+            textfont=dict(size=14, color="white")  # Adjust text color for visibility
+        )
+    else:
+        _fig.update_traces(
+            textposition="outside",
+            # error_y=dict(thickness=0)
+        )
+
+    # * set axis title
+
     _fig.show(renderer)
 
     # * save to png if path is provided

pandas_plots/tbl.py

@@ -269,6 +269,7 @@ def pivot_df(
     alter_font: bool = True,
     font_size_th: int = 0,
     font_size_td: int = 0,
+    col1_width: int = 0,
     png_path: str | Path = None,
     png_conversion: Literal["chrome", "selenium"] = "selenium",
 ) -> pd.DataFrame:
@@ -306,6 +307,7 @@ def pivot_df(
         alter_font (bool, optional): Whether to alter the font. Defaults to True.
         font_size_th (int, optional): The font size for the header. Defaults to 0.
         font_size_td (int, optional): The font size for the table data. Defaults to 0.
+        col1_width (int, optional): The width of the first column in px. Defaults to 0.
         png_path (str | Path, optional): The path to save the output PNG file. Defaults to None.
         png_conversion (Literal["chrome", "selenium"], optional): The conversion method for the PNG file. Defaults to "selenium".
 
@@ -393,6 +395,7 @@ def pivot_df(
         alter_font=alter_font,
         font_size_th=font_size_th,
         font_size_td=font_size_td,
+        col1_width=col1_width,
         png_path=png_path,
         png_conversion=png_conversion,
         
@@ -416,6 +419,7 @@ def show_num_df(
     alter_font: bool = True,
     font_size_th: int = 0,
     font_size_td: int = 0,
+    col1_width: int = 0,
     png_path: str | Path = None,
     png_conversion: Literal["chrome", "selenium"] = "selenium",
 ):
@@ -445,6 +449,7 @@ def show_num_df(
     - alter_font: a boolean indicating whether to alter the font family
     - font_size_th: an integer indicating the font size for the header
     - font_size_td: an integer indicating the font size for the table data
+    - col1_width: an integer indicating the width of the first column in px
     - png_path: a string or Path indicating the path to save the PNG file
     - png_conversion: a Literal indicating the conversion method for the PNG file ["chrome", "selenium"]
 
@@ -680,6 +685,15 @@ def show_num_df(
             dict(selector="td", props=_props_td),
         ]
     )
+    
+    if  col1_width > 0:
+        out.set_table_styles([
+            {'selector': 'th:first-child, td:first-child', 
+            'props': [(f'min-width', f'{col1_width}px !important'), 
+                    (f'max-width', f'{col1_width}px !important'), 
+                    ('white-space', 'nowrap'), 
+                    ('overflow', 'hidden')]}
+        ])
 
     if heatmap_axis:
         out.background_gradient(

{pandas_plots-0.12.16.dist-info → pandas_plots-0.12.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pandas-plots
-Version: 0.12.16
+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.18.dist-info/RECORD

@@ -0,0 +1,11 @@
+pandas_plots/hlp.py,sha256=hlz7kKe6iDsz6Ov5YadX9zT5E01DHy7gHdVPgG8P7nQ,19656
+pandas_plots/pii.py,sha256=2WKE-W9s285jPdsTqCgt1uxuW4lj1PYCVOYB2fYDNwQ,2195
+pandas_plots/pls.py,sha256=YbWPk34oJ3SJ-n7O-FvMOYR2FT2uNbMe93E0dbVTetk,48170
+pandas_plots/tbl.py,sha256=LxMKJh4qkGuQZ1DdCZIq1tMS26F6elsqbe_uabvQx4E,32535
+pandas_plots/ven.py,sha256=2x3ACo2vSfO3q6fv-UdDQ0h1SJyt8WChBGgE5SDCdCk,11673
+pandas_plots-0.12.18.dist-info/LICENSE,sha256=6KQ5KVAAhRaB-JJKpX4cefKvRZRgI7GUPc92_2d31XY,1051
+pandas_plots-0.12.18.dist-info/METADATA,sha256=0JzJDbgvUgQAIsn3X3bj4B1ZVaU2L1U0SegY2OVgEH8,7550
+pandas_plots-0.12.18.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+pandas_plots-0.12.18.dist-info/pii.py,sha256=2WKE-W9s285jPdsTqCgt1uxuW4lj1PYCVOYB2fYDNwQ,2195
+pandas_plots-0.12.18.dist-info/top_level.txt,sha256=XnaNuIHBqMmCeh_U7nKOYTwFue_SIA0wxuDgdPmnnSk,13
+pandas_plots-0.12.18.dist-info/RECORD,,

pandas_plots-0.12.16.dist-info/RECORD

@@ -1,11 +0,0 @@
-pandas_plots/hlp.py,sha256=ggIe9uwF4tlVWqaBxQOkn8Dz8955xjY-TYG-lzypWFE,18905
-pandas_plots/pii.py,sha256=2WKE-W9s285jPdsTqCgt1uxuW4lj1PYCVOYB2fYDNwQ,2195
-pandas_plots/pls.py,sha256=OLaX5gWbsuYniikpojqPxZt9oQK_PI1XyJEQkbNlSk8,47809
-pandas_plots/tbl.py,sha256=VNhb-3Vezl4imIS5omzHFd9rsEYFwq2YIx-DFVtiOSc,31930
-pandas_plots/ven.py,sha256=2x3ACo2vSfO3q6fv-UdDQ0h1SJyt8WChBGgE5SDCdCk,11673
-pandas_plots-0.12.16.dist-info/LICENSE,sha256=6KQ5KVAAhRaB-JJKpX4cefKvRZRgI7GUPc92_2d31XY,1051
-pandas_plots-0.12.16.dist-info/METADATA,sha256=rLLfHJM27ITmbsVzso-HKiEaPflY-pTpAt31xiORPWA,7550
-pandas_plots-0.12.16.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-pandas_plots-0.12.16.dist-info/pii.py,sha256=2WKE-W9s285jPdsTqCgt1uxuW4lj1PYCVOYB2fYDNwQ,2195
-pandas_plots-0.12.16.dist-info/top_level.txt,sha256=XnaNuIHBqMmCeh_U7nKOYTwFue_SIA0wxuDgdPmnnSk,13
-pandas_plots-0.12.16.dist-info/RECORD,,