pandas-plots 0.12.3__py3-none-any.whl → 0.12.5__py3-none-any.whl

pandas_plots/pls.py

@@ -78,7 +78,7 @@ def assign_column_colors(columns, color_palette, null_label):
         raise ValueError(f"Invalid color palette: {color_palette}")
     
     colors = {col: palette[i % len(palette)] for i, col in enumerate(sorted(columns))}
-    colors[null_label] = "gray"
+    colors[null_label] = "lightgray"
     return colors
 
 ### main functions
@@ -192,12 +192,32 @@ def plot_stacked_bars(
 ) -> object:
     """
     Generates a stacked bar plot using the provided DataFrame.
-    Updated to assign colors using `assign_column_colors` with nulls colored grey.
 
     Parameters:
-    All parameters are similar to the original function, with the addition of:
-    - color_palette: str - Name of the color palette.
-    - null_label: str - Label for null values.
+    - df (pd.DataFrame): The input DataFrame with at least two categorical columns and one numerical column.
+    - top_n_index (int): Limit the number of categories displayed on the index axis.
+    - top_n_color (int): Limit the number of categories displayed in the color legend.
+    - dropna (bool): If True, removes rows with missing values; otherwise, replaces them with `null_label`.
+    - swap (bool): If True, swaps the first two columns.
+    - normalize (bool): If True, normalizes numerical values between 0 and 1.
+    - relative (bool): If True, normalizes the bars to a percentage scale.
+    - orientation (Literal["h", "v"]): Defines the orientation of the bars ("v" for vertical, "h" for horizontal).
+    - height (int): Height of the plot.
+    - width (int): Width of the plot.
+    - title (str): Custom title for the plot.
+    - renderer (Literal["png", "svg", None]): Defines the output format.
+    - caption (str): Optional caption for additional context.
+    - sort_values (bool): 
+        - If True, sorts bars by the sum of their values (descending).
+        - If False, sorts bars alphabetically.
+    - show_total (bool): If True, adds a row with the total sum of all categories.
+    - precision (int): Number of decimal places for numerical values.
+    - png_path (Path | str): If specified, saves the plot as a PNG file.
+    - color_palette (str): Name of the color palette to use.
+    - null_label (str): Label for null values.
+    
+    Returns:
+    - A Plotly figure object representing the stacked bar chart.
     """
     BAR_LENGTH_MULTIPLIER = 1.05
 
@@ -211,6 +231,8 @@ def plot_stacked_bars(
         print("❌ first 2 columns must be str")
         return
 
+    df = df.copy()  # Copy the input DataFrame to avoid modifying the original
+
     # * add count column[2] as a service if none is present
     if len(df.columns) == 2:
         df["cnt"] = 1
@@ -251,6 +273,18 @@ def plot_stacked_bars(
         .sum()
         .reset_index()
     )
+    
+    # * Sorting logic based on sort_values
+    if sort_values:
+        sort_order = (
+            df.groupby(col_index)[df.columns[2]].sum().sort_values(ascending=False).index
+        )
+    else:
+        sort_order = sorted(df[col_index].unique())  # Alphabetical order
+
+    # * Convert to categorical with explicit ordering
+    df[col_index] = pd.Categorical(df[col_index], categories=sort_order, ordered=True)
+
 
     # * calculate n
     divider = 2 if show_total else 1
@@ -264,7 +298,7 @@ def plot_stacked_bars(
     caption = _set_caption(caption)
 
     # * plot
-    _fig = px.bar(
+    fig = px.bar(
         df,
         x=col_index if orientation == "v" else df.columns[2],
         y=df.columns[2] if orientation == "v" else col_index,
@@ -277,20 +311,58 @@ def plot_stacked_bars(
         width=width,
         height=height,
         color_discrete_map=column_colors,  # Use assigned colors
+        category_orders={col_index: list(df[col_index].cat.categories)},  # <- Add this line
     )
+        # * get longest bar
+    bar_max = (
+        df.groupby(col_index)[df.columns[2]].sum().sort_values(ascending=False).iloc[0]
+        * BAR_LENGTH_MULTIPLIER
+    )
+    # * ignore if bar mode is on
+    if not relative:
+        if orientation == "v":
+            fig.update_yaxes(range=[0, bar_max])
+        else:
+            fig.update_xaxes(range=[0, bar_max])
+    else:
+        fig.update_layout(barnorm="percent")
+
+    # * set title properties
+    fig.update_layout(
+        title={
+            # 'x': 0.1,
+            "y": 0.95,
+            "xanchor": "left",
+            "yanchor": "top",
+            "font": {
+                "size": 24,
+            },
+        },
+    )
+    
+    # * set dtick
+    if orientation == "h":
+        if relative:
+            fig.update_xaxes(dtick=5)
+        elif normalize:
+            fig.update_xaxes(dtick=0.05)
+    else:
+        if relative:
+            fig.update_yaxes(dtick=5)
+        elif normalize:
+            fig.update_yaxes(dtick=0.05)
 
     # * show grids, set to smaller distance on pct scale
-    _fig.update_xaxes(showgrid=True, gridwidth=1)
-    _fig.update_yaxes(showgrid=True, gridwidth=1)
+    fig.update_xaxes(showgrid=True, gridwidth=1)
+    fig.update_yaxes(showgrid=True, gridwidth=1)
 
     # * save to png if path is provided
     if png_path is not None:
-        _fig.write_image(Path(png_path).as_posix())
-
-    _fig.show(renderer)
+        fig.write_image(Path(png_path).as_posix())
 
-    return _fig
+    fig.show(renderer=renderer)
 
+    return fig
 
 
 def plot_bars(
@@ -1151,17 +1223,17 @@ def plot_facet_stacked_bars(
     unique_rows = len(aggregated_df)
     axis_details = []
     if top_n_index > 0:
-        axis_details.append(f"top {top_n_index} [{original_column_names[0]}]")
+        axis_details.append(f"TOP {top_n_index} [{original_column_names[0]}]")
     else:
         axis_details.append(f"[{original_column_names[0]}]")
 
     if top_n_columns > 0:
-        axis_details.append(f"top {top_n_columns} [{original_column_names[1]}]")
+        axis_details.append(f"TOP {top_n_columns} [{original_column_names[1]}]")
     else:
         axis_details.append(f"[{original_column_names[1]}]")
 
     if top_n_facet > 0:
-        axis_details.append(f"top {top_n_facet} [{original_column_names[2]}]")
+        axis_details.append(f"TOP {top_n_facet} [{original_column_names[2]}]")
     else:
         axis_details.append(f"[{original_column_names[2]}]")
 
@@ -1180,6 +1252,6 @@ def plot_facet_stacked_bars(
         png_path = Path(png_path)
         fig.write_image(str(png_path))
 
-    fig.show(renderer)
+    fig.show(renderer=renderer)
 
     return fig

{pandas_plots-0.12.3.dist-info → pandas_plots-0.12.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pandas-plots
-Version: 0.12.3
+Version: 0.12.5
 Summary: A collection of helper for table handling and visualization
 Home-page: https://github.com/smeisegeier/pandas-plots
 Author: smeisegeier
@@ -20,7 +20,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pandas>=2.0.0
-Requires-Dist: plotly>=5.18.0
+Requires-Dist: plotly<6
 Requires-Dist: matplotlib>=3.8.2
 Requires-Dist: matplotlib-venn==0.11.10
 Requires-Dist: seaborn>=0.13.2
@@ -96,7 +96,7 @@ tbl.show_num_df(
   - `plot_histogram()` histogram for one or more **numerical** columns
   - `plot_joints()` a joint plot for **exactly two numerical** columns
   - `plot_quadrants()` quickly shows a 2x2 heatmap
-  - 🆕 `plot_stacked_bars()` shows stacked bars for a facet value as subplots 
+  - 🆕 `plot_facet_stacked_bars()` shows stacked bars for a facet value as subplots 
 <br>
 
 - `ven` offers functions for _venn diagrams_

pandas_plots-0.12.5.dist-info/RECORD

@@ -0,0 +1,10 @@
+pandas_plots/hlp.py,sha256=N6NrbFagVMMX-ZnV0rIBEz82SeSoOkksfMcCap55W7E,16588
+pandas_plots/pii.py,sha256=2WKE-W9s285jPdsTqCgt1uxuW4lj1PYCVOYB2fYDNwQ,2195
+pandas_plots/pls.py,sha256=L10KeIvN1sNWtC6wK5IehinIQzcDVDrgx3DHxTy3cnU,42136
+pandas_plots/tbl.py,sha256=4VvjLisPT1gSvgsLClcrhC7LIJ-_FPNla8nomGflGag,30509
+pandas_plots/ven.py,sha256=2x3ACo2vSfO3q6fv-UdDQ0h1SJyt8WChBGgE5SDCdCk,11673
+pandas_plots-0.12.5.dist-info/LICENSE,sha256=6KQ5KVAAhRaB-JJKpX4cefKvRZRgI7GUPc92_2d31XY,1051
+pandas_plots-0.12.5.dist-info/METADATA,sha256=_rqHgKQ3vHsRFb57i0i5ZYG3SugslGUeVudP0pLtl2w,7358
+pandas_plots-0.12.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+pandas_plots-0.12.5.dist-info/top_level.txt,sha256=XnaNuIHBqMmCeh_U7nKOYTwFue_SIA0wxuDgdPmnnSk,13
+pandas_plots-0.12.5.dist-info/RECORD,,

pandas_plots-0.12.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-pandas_plots/hlp.py,sha256=N6NrbFagVMMX-ZnV0rIBEz82SeSoOkksfMcCap55W7E,16588
-pandas_plots/pii.py,sha256=2WKE-W9s285jPdsTqCgt1uxuW4lj1PYCVOYB2fYDNwQ,2195
-pandas_plots/pls.py,sha256=nNpAPIxONcoUb_E6U6oGEEjva6MFgycc3pBq0fb1TQ0,39268
-pandas_plots/tbl.py,sha256=4VvjLisPT1gSvgsLClcrhC7LIJ-_FPNla8nomGflGag,30509
-pandas_plots/ven.py,sha256=2x3ACo2vSfO3q6fv-UdDQ0h1SJyt8WChBGgE5SDCdCk,11673
-pandas_plots-0.12.3.dist-info/LICENSE,sha256=6KQ5KVAAhRaB-JJKpX4cefKvRZRgI7GUPc92_2d31XY,1051
-pandas_plots-0.12.3.dist-info/METADATA,sha256=B_W5POVonGswuIrADMZGWiKNVuwWkXBKQN9CQj8-wP8,7358
-pandas_plots-0.12.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-pandas_plots-0.12.3.dist-info/top_level.txt,sha256=XnaNuIHBqMmCeh_U7nKOYTwFue_SIA0wxuDgdPmnnSk,13
-pandas_plots-0.12.3.dist-info/RECORD,,