flatbread 0.1.0__tar.gz → 0.1.1__tar.gz

{flatbread-0.1.0 → flatbread-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flatbread
-Version: 0.1.0
+Version: 0.1.1
 Summary: Pandas extension for aggregation and tabular display
 Project-URL: Homepage, https://github.com/lcvriend/flatbread
 Author-email: "L.C. Vriend" <vanboefer@gmail.com>
@@ -695,20 +695,18 @@ Requires-Dist: jinja2
 Requires-Dist: pandas>=2.0.0
 Description-Content-Type: text/markdown
 
-Here's a concise README for flatbread:
-
-```markdown
 # Flatbread
 
-Flatbread is a Python library that extends pandas with tabulation features through a chainable API. It makes it easy to create summary tables with totals, subtotals, percentages, and aggregations.
+Flatbread is a Python library that extends pandas with tabulation features. It makes it easy to create and display summary tables with totals, subtotals, percentages, and aggregations.
 
 Flatbread can be accessed in `DataFrames` and `Series` using the `pita` accessor.
 
+It uses the [wc-simple-table](https://github.com/lcvriend/wc-simple-table) dataviewer web component to display tables in a notebook: [check out some examples](https://lcvriend.github.io/wc-simple-table/).
+
 ## Key Features
 
 - Add row and column totals/subtotals to DataFrames and Series
 - Calculate and format percentages with proper rounding
-- Chain operations
 - Preserve data types and index structures
 - Table display in Jupyter notebooks
 
@@ -720,12 +718,12 @@ import flatbread
 
 df = pd.DataFrame(...)
 
-# Add totals and percentages in one chain
+# Add totals and percentages
 result = (
-    df.pita
-    .add_totals()           # Add grand totals
-    .add_subtotals(level=0) # Add subtotals by first index level
-    .add_percentages()      # Add percentage columns
+    df
+    .pita.add_totals()           # Add grand totals
+    .pita.add_subtotals(level=0) # Add subtotals by first index level
+    .pita.add_percentages()      # Add percentage columns
 )
 
 # Display with interactive viewer

{flatbread-0.1.0 → flatbread-0.1.1}/flatbread/tooling.py

@@ -86,7 +86,7 @@ def sort_totals(
 @singledispatch
 def add_level(
     data,
-    item: Any,
+    values: Any|list[Any],
     level: int = 0,
     level_name: Any = None,
     axis: Axis = 0,
@@ -97,7 +97,7 @@ def add_level(
 @add_level.register
 def _(
     data: pd.DataFrame,
-    value: Any,
+    value: Any|list[Any],
     level: int = 0,
     level_name: Any = None,
     axis: Axis = 0,
@@ -109,8 +109,8 @@ def _(
     ----------
     data (pd.DataFrame):
         Input DataFrame.
-    value (Any):
-        Value to fill the new level with.
+    value Any|list[Any]
+        Either a single value to fill the entire level with, or a list of values with length matching the axis size. Values will be mapped in order.
     level (int, optional):
         Position to insert the new level. Defaults to 0 (start).
     level_name (Any, optional):
@@ -126,11 +126,25 @@ def _(
     data = data.copy()
     target = data.index if axis in [0, 'index'] else data.columns
 
+    if isinstance(value, list):
+        if len(value) != len(target):
+            raise ValueError(
+                f"Length of values list ({len(value)}) must match "
+                f"length of {'index' if axis in [0, 'index'] else 'columns'} ({len(target)})"
+            )
+
     if not isinstance(target, pd.MultiIndex):
         original_name = target.name
         target = pd.MultiIndex.from_arrays([target], names=[original_name])
 
-    new_keys = [add_value_to_key(key, value, level) for key in target]
+    new_keys = [
+        add_value_to_key(
+            key,
+            value[i] if isinstance(value, list) else value,
+            level
+        )
+        for i, key in enumerate(target)
+    ]
     new_names = add_value_to_key(target.names, level_name, level)
     new_index = pd.MultiIndex.from_tuples(new_keys, names=new_names)
 
@@ -156,8 +170,8 @@ def _(
     ----------
     data (pd.Series):
         Input Series.
-    value (Any):
-        Value to fill the new level with.
+    value Any|list[Any]
+        Either a single value to fill the entire level with, or a list of values with length matching the axis size. Values will be mapped in order.
     level (int, optional):
         Position to insert the new level. Defaults to 0 (start).
     level_name (Any, optional):
@@ -173,11 +187,25 @@ def _(
     data = data.copy()
     target = data.index
 
+    if isinstance(value, list):
+        if len(value) != len(target):
+            raise ValueError(
+                f"Length of values list ({len(value)}) must match "
+                f"length of index ({len(target)})"
+            )
+
     if not isinstance(target, pd.MultiIndex):
         original_name = target.name
         target = pd.MultiIndex.from_arrays([target], names=[original_name])
 
-    new_keys = [add_value_to_key(key, value, level) for key in target]
+    new_keys = [
+        add_value_to_key(
+            key,
+            value[i] if isinstance(value, list) else value,
+            level
+        )
+        for i, key in enumerate(target)
+    ]
     new_names = add_value_to_key(target.names, level_name, level)
     new_index = pd.MultiIndex.from_tuples(new_keys, names=new_names)
     data.index = new_index

{flatbread-0.1.0 → flatbread-0.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "flatbread"
-version = "0.1.0"
+version = "0.1.1"
 description = "Pandas extension for aggregation and tabular display"
 readme = "readme.md"
 requires-python = ">=3.10"

{flatbread-0.1.0 → flatbread-0.1.1}/readme.md

@@ -1,17 +1,15 @@
-Here's a concise README for flatbread:
-
-```markdown
 # Flatbread
 
-Flatbread is a Python library that extends pandas with tabulation features through a chainable API. It makes it easy to create summary tables with totals, subtotals, percentages, and aggregations.
+Flatbread is a Python library that extends pandas with tabulation features. It makes it easy to create and display summary tables with totals, subtotals, percentages, and aggregations.
 
 Flatbread can be accessed in `DataFrames` and `Series` using the `pita` accessor.
 
+It uses the [wc-simple-table](https://github.com/lcvriend/wc-simple-table) dataviewer web component to display tables in a notebook: [check out some examples](https://lcvriend.github.io/wc-simple-table/).
+
 ## Key Features
 
 - Add row and column totals/subtotals to DataFrames and Series
 - Calculate and format percentages with proper rounding
-- Chain operations
 - Preserve data types and index structures
 - Table display in Jupyter notebooks
 
@@ -23,12 +21,12 @@ import flatbread
 
 df = pd.DataFrame(...)
 
-# Add totals and percentages in one chain
+# Add totals and percentages
 result = (
-    df.pita
-    .add_totals()           # Add grand totals
-    .add_subtotals(level=0) # Add subtotals by first index level
-    .add_percentages()      # Add percentage columns
+    df
+    .pita.add_totals()           # Add grand totals
+    .pita.add_subtotals(level=0) # Add subtotals by first index level
+    .pita.add_percentages()      # Add percentage columns
 )
 
 # Display with interactive viewer