xarray-plotly 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

xarray_plotly/accessor.py

@@ -34,7 +34,7 @@ class DataArrayPlotlyAccessor:
         ```
     """
 
-    __all__: ClassVar = ["line", "bar", "area", "scatter", "box", "imshow", "pie"]
+    __all__: ClassVar = ["line", "bar", "fast_bar", "area", "scatter", "box", "imshow", "pie"]
 
     def __init__(self, darray: DataArray) -> None:
         self._da = darray
@@ -160,6 +160,41 @@ class DataArrayPlotlyAccessor:
             **px_kwargs,
         )
 
+    def fast_bar(
+        self,
+        *,
+        x: SlotValue = auto,
+        color: SlotValue = auto,
+        facet_col: SlotValue = auto,
+        facet_row: SlotValue = auto,
+        animation_frame: SlotValue = auto,
+        **px_kwargs: Any,
+    ) -> go.Figure:
+        """Create a bar-like chart using stacked areas for better performance.
+
+        Slot order: x -> color -> facet_col -> facet_row -> animation_frame
+
+        Args:
+            x: Dimension for x-axis. Default: first dimension.
+            color: Dimension for color/stacking. Default: second dimension.
+            facet_col: Dimension for subplot columns. Default: third dimension.
+            facet_row: Dimension for subplot rows. Default: fourth dimension.
+            animation_frame: Dimension for animation. Default: fifth dimension.
+            **px_kwargs: Additional arguments passed to `plotly.express.area()`.
+
+        Returns:
+            Interactive Plotly Figure.
+        """
+        return plotting.fast_bar(
+            self._da,
+            x=x,
+            color=color,
+            facet_col=facet_col,
+            facet_row=facet_row,
+            animation_frame=animation_frame,
+            **px_kwargs,
+        )
+
     def scatter(
         self,
         *,
@@ -349,7 +384,7 @@ class DatasetPlotlyAccessor:
         ```
     """
 
-    __all__: ClassVar = ["line", "bar", "area", "scatter", "box", "pie"]
+    __all__: ClassVar = ["line", "bar", "fast_bar", "area", "scatter", "box", "pie"]
 
     def __init__(self, dataset: Dataset) -> None:
         self._ds = dataset
@@ -501,6 +536,42 @@ class DatasetPlotlyAccessor:
             **px_kwargs,
         )
 
+    def fast_bar(
+        self,
+        var: str | None = None,
+        *,
+        x: SlotValue = auto,
+        color: SlotValue = auto,
+        facet_col: SlotValue = auto,
+        facet_row: SlotValue = auto,
+        animation_frame: SlotValue = auto,
+        **px_kwargs: Any,
+    ) -> go.Figure:
+        """Create a bar-like chart using stacked areas for better performance.
+
+        Args:
+            var: Variable to plot. If None, plots all variables with "variable" dimension.
+            x: Dimension for x-axis.
+            color: Dimension for color/stacking.
+            facet_col: Dimension for subplot columns.
+            facet_row: Dimension for subplot rows.
+            animation_frame: Dimension for animation.
+            **px_kwargs: Additional arguments passed to `plotly.express.area()`.
+
+        Returns:
+            Interactive Plotly Figure.
+        """
+        da = self._get_dataarray(var)
+        return plotting.fast_bar(
+            da,
+            x=x,
+            color=color,
+            facet_col=facet_col,
+            facet_row=facet_row,
+            animation_frame=animation_frame,
+            **px_kwargs,
+        )
+
     def scatter(
         self,
         var: str | None = None,

xarray_plotly/config.py

@@ -26,6 +26,7 @@ DEFAULT_SLOT_ORDERS: dict[str, tuple[str, ...]] = {
         "animation_frame",
     ),
     "bar": ("x", "color", "pattern_shape", "facet_col", "facet_row", "animation_frame"),
+    "fast_bar": ("x", "color", "facet_col", "facet_row", "animation_frame"),
     "area": (
         "x",
         "color",

xarray_plotly/figures.py

@@ -8,9 +8,28 @@ import copy
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from collections.abc import Iterator
+
     import plotly.graph_objects as go
 
 
+def _iter_all_traces(fig: go.Figure) -> Iterator:
+    """Iterate over all traces in a figure, including animation frames.
+
+    Yields traces from fig.data first, then from each frame in fig.frames.
+    Useful for applying styling to all traces including those in animations.
+
+    Args:
+        fig: Plotly Figure.
+
+    Yields:
+        Each trace object from the figure.
+    """
+    yield from fig.data
+    for frame in fig.frames or []:
+        yield from frame.data
+
+
 def _get_subplot_axes(fig: go.Figure) -> set[tuple[str, str]]:
     """Extract (xaxis, yaxis) pairs from figure traces.
 
@@ -418,15 +437,12 @@ def update_traces(fig: go.Figure, selector: dict | None = None, **kwargs) -> go.
         >>> # Update specific trace by name
         >>> update_traces(fig, selector={"name": "Germany"}, line_width=5, line_dash="dot")
     """
-    fig.update_traces(selector=selector, **kwargs)
-
-    for frame in fig.frames:
-        for trace in frame.data:
-            if selector is None:
+    for trace in _iter_all_traces(fig):
+        if selector is None:
+            trace.update(**kwargs)
+        else:
+            # Check if trace matches all selector criteria
+            if all(getattr(trace, k, None) == v for k, v in selector.items()):
                 trace.update(**kwargs)
-            else:
-                # Check if trace matches all selector criteria
-                if all(getattr(trace, k, None) == v for k, v in selector.items()):
-                    trace.update(**kwargs)
 
     return fig

xarray_plotly/plotting.py

@@ -4,6 +4,7 @@ Plotly Express plotting functions for DataArray objects.
 
 from __future__ import annotations
 
+import warnings
 from typing import TYPE_CHECKING, Any
 
 import numpy as np
@@ -18,6 +19,9 @@ from xarray_plotly.common import (
     get_value_col,
     to_dataframe,
 )
+from xarray_plotly.figures import (
+    _iter_all_traces,
+)
 
 if TYPE_CHECKING:
     import plotly.graph_objects as go
@@ -167,6 +171,167 @@ def bar(
     )
 
 
+def _classify_trace_sign(y_values: np.ndarray) -> str:
+    """Classify a trace as 'positive', 'negative', or 'mixed' based on its values."""
+    y_arr = np.asarray(y_values)
+    y_clean = y_arr[np.isfinite(y_arr) & (np.abs(y_arr) > 1e-9)]
+    if len(y_clean) == 0:
+        return "zero"
+    has_pos = bool(np.any(y_clean > 0))
+    has_neg = bool(np.any(y_clean < 0))
+    if has_pos and has_neg:
+        return "mixed"
+    elif has_neg:
+        return "negative"
+    elif has_pos:
+        return "positive"
+    return "zero"
+
+
+def _style_traces_as_bars(fig: go.Figure) -> None:
+    """Style area chart traces to look like bar charts with proper pos/neg stacking.
+
+    Classifies each trace (by name) across all data and animation frames,
+    then assigns stackgroups: positive traces stack upward, negative stack downward.
+    """
+    # Collect all traces (main + animation frames)
+    all_traces = list(_iter_all_traces(fig))
+
+    # Classify each trace name by aggregating sign info across all occurrences
+    sign_flags: dict[str, dict[str, bool]] = {}
+    for trace in all_traces:
+        if trace.name not in sign_flags:
+            sign_flags[trace.name] = {"has_pos": False, "has_neg": False}
+        if trace.y is not None and len(trace.y) > 0:
+            y_arr = np.asarray(trace.y)
+            y_clean = y_arr[np.isfinite(y_arr) & (np.abs(y_arr) > 1e-9)]
+            if len(y_clean) > 0:
+                if np.any(y_clean > 0):
+                    sign_flags[trace.name]["has_pos"] = True
+                if np.any(y_clean < 0):
+                    sign_flags[trace.name]["has_neg"] = True
+
+    # Build classification map
+    class_map: dict[str, str] = {}
+    mixed_traces: list[str] = []
+    for name, flags in sign_flags.items():
+        if flags["has_pos"] and flags["has_neg"]:
+            class_map[name] = "mixed"
+            mixed_traces.append(name)
+        elif flags["has_neg"]:
+            class_map[name] = "negative"
+        elif flags["has_pos"]:
+            class_map[name] = "positive"
+        else:
+            class_map[name] = "zero"
+
+    # Warn about mixed traces
+    if mixed_traces:
+        warnings.warn(
+            f"fast_bar: traces {mixed_traces} have mixed positive/negative values "
+            "and cannot be stacked. They are shown as dashed lines. "
+            "Consider using bar() for proper stacking of mixed data.",
+            UserWarning,
+            stacklevel=3,
+        )
+
+    # Apply styling to all traces
+    for trace in all_traces:
+        color = trace.line.color
+        cls = class_map.get(trace.name, "positive")
+
+        if cls in ("positive", "negative"):
+            trace.stackgroup = cls
+            trace.fillcolor = color
+            trace.line = {"width": 0, "color": color, "shape": "hv"}
+        elif cls == "mixed":
+            # Mixed: no stacking, show as dashed line
+            trace.stackgroup = None
+            trace.fill = None
+            trace.line = {"width": 2, "color": color, "shape": "hv", "dash": "dash"}
+        else:  # zero
+            trace.stackgroup = None
+            trace.fill = None
+            trace.line = {"width": 0, "color": color, "shape": "hv"}
+
+
+def fast_bar(
+    darray: DataArray,
+    *,
+    x: SlotValue = auto,
+    color: SlotValue = auto,
+    facet_col: SlotValue = auto,
+    facet_row: SlotValue = auto,
+    animation_frame: SlotValue = auto,
+    **px_kwargs: Any,
+) -> go.Figure:
+    """
+    Create a bar-like chart using stacked areas for better performance.
+
+    Uses `px.area` with stepped lines and no outline to create a bar-like
+    appearance. Renders faster than `bar()` for large datasets because it
+    uses a single polygon per trace instead of individual rectangles.
+
+    The y-axis shows DataArray values. Dimensions fill slots in order:
+    x -> color -> facet_col -> facet_row -> animation_frame
+
+    Traces are classified by their values: purely positive traces stack upward,
+    purely negative traces stack downward. Traces with mixed signs are shown
+    as dashed lines without stacking.
+
+    Parameters
+    ----------
+    darray
+        The DataArray to plot.
+    x
+        Dimension for x-axis. Default: first dimension.
+    color
+        Dimension for color/stacking. Default: second dimension.
+    facet_col
+        Dimension for subplot columns. Default: third dimension.
+    facet_row
+        Dimension for subplot rows. Default: fourth dimension.
+    animation_frame
+        Dimension for animation. Default: fifth dimension.
+    **px_kwargs
+        Additional arguments passed to `plotly.express.area()`.
+
+    Returns
+    -------
+    plotly.graph_objects.Figure
+    """
+    slots = assign_slots(
+        list(darray.dims),
+        "fast_bar",
+        x=x,
+        color=color,
+        facet_col=facet_col,
+        facet_row=facet_row,
+        animation_frame=animation_frame,
+    )
+
+    df = to_dataframe(darray)
+    value_col = get_value_col(darray)
+    labels = {**build_labels(darray, slots, value_col), **px_kwargs.pop("labels", {})}
+
+    fig = px.area(
+        df,
+        x=slots.get("x"),
+        y=value_col,
+        color=slots.get("color"),
+        facet_col=slots.get("facet_col"),
+        facet_row=slots.get("facet_row"),
+        animation_frame=slots.get("animation_frame"),
+        line_shape="hv",
+        labels=labels,
+        **px_kwargs,
+    )
+
+    _style_traces_as_bars(fig)
+
+    return fig
+
+
 def area(
     darray: DataArray,
     *,

{xarray_plotly-0.0.7.dist-info → xarray_plotly-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xarray_plotly
-Version: 0.0.7
+Version: 0.0.8
 Summary: Interactive Plotly Express plotting accessor for xarray
 Author: Felix
 License: MIT

xarray_plotly-0.0.8.dist-info/RECORD

@@ -0,0 +1,12 @@
+xarray_plotly/__init__.py,sha256=vAM2TCeLnSkSjwCDETlBI_1SukhVeth7HFMaFIWl_ps,3384
+xarray_plotly/accessor.py,sha256=eP4_GPBCN-c32M8s1LJv4Zn4PvHHO9HsP5oPQMDDd6I,23150
+xarray_plotly/common.py,sha256=YTiaPLJ0Gh20mHV8-72J8DjWk-XaSYaT_ZiqXp6cecU,7192
+xarray_plotly/config.py,sha256=gS6IqWdx82PQs1yZzl_diGHlnmVTladuWZzk1lYgIsQ,6569
+xarray_plotly/figures.py,sha256=5SxTTPCgaEMRC5Bz7Ew1leJXDNMA1ZchbY_dO82SHY4,15574
+xarray_plotly/plotting.py,sha256=Fl_UklCN8AYtPj0t3m8zXwcz6MsFLdmQrKZC68IH1OM,20668
+xarray_plotly/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xarray_plotly-0.0.8.dist-info/licenses/LICENSE,sha256=AvVEfNqbhIm9jHvt0acJNjW1JUKa2a70Zb5rJdEXCJI,1064
+xarray_plotly-0.0.8.dist-info/METADATA,sha256=ezKDOXZ7CklqOB5KDBh32Lm1nri63OF_OoeCKhuIt9c,3415
+xarray_plotly-0.0.8.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+xarray_plotly-0.0.8.dist-info/top_level.txt,sha256=GtMkvuZvLAYTjYXtwoNUa0ag42CJARZJK1CZemYD7pg,14
+xarray_plotly-0.0.8.dist-info/RECORD,,

xarray_plotly-0.0.7.dist-info/RECORD

@@ -1,12 +0,0 @@
-xarray_plotly/__init__.py,sha256=vAM2TCeLnSkSjwCDETlBI_1SukhVeth7HFMaFIWl_ps,3384
-xarray_plotly/accessor.py,sha256=rbVqzi29nvQUM8b5jE5ivw-CPoUxQjrl5Fnh-LQvtWo,20758
-xarray_plotly/common.py,sha256=YTiaPLJ0Gh20mHV8-72J8DjWk-XaSYaT_ZiqXp6cecU,7192
-xarray_plotly/config.py,sha256=-Sp6TNx-8Zk6x0WmJ5j_KGHPRY639ju3_RSemrlmuXY,6492
-xarray_plotly/figures.py,sha256=fE6aLqcNuCZ5GA-3x4xTl1Bw9Fg9fGCoyjKdE7jM-eo,15161
-xarray_plotly/plotting.py,sha256=i06kz7HSloH_SfkmkjRnafz1auADqFi7X5N3tZKZ1_E,15239
-xarray_plotly/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xarray_plotly-0.0.7.dist-info/licenses/LICENSE,sha256=AvVEfNqbhIm9jHvt0acJNjW1JUKa2a70Zb5rJdEXCJI,1064
-xarray_plotly-0.0.7.dist-info/METADATA,sha256=6YUJZEytGX_sjeAXz-jeW9Wpuct2yicNBeo8bauGemc,3415
-xarray_plotly-0.0.7.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-xarray_plotly-0.0.7.dist-info/top_level.txt,sha256=GtMkvuZvLAYTjYXtwoNUa0ag42CJARZJK1CZemYD7pg,14
-xarray_plotly-0.0.7.dist-info/RECORD,,