xarray-plotly 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl

xarray_plotly/common.py

@@ -13,7 +13,7 @@ from xarray_plotly.config import DEFAULT_SLOT_ORDERS, _options
 
 if TYPE_CHECKING:
     import pandas as pd
-    from xarray import DataArray
+    from xarray import DataArray, Dataset
 
 
 class _AUTO:
@@ -251,7 +251,21 @@ def _get_qualitative_scale_names() -> frozenset[str]:
     )
 
 
-def resolve_colors(colors: Colors, px_kwargs: dict[str, Any]) -> dict[str, Any]:
+def _sample_colorscale(name: str, n: int) -> list[str]:
+    """Sample *n* evenly-spaced colors from a named Plotly colorscale."""
+    scale = px.colors.get_colorscale(name)
+    samplepoints = [i / max(n - 1, 1) for i in range(n)]
+    result: list[str] = px.colors.sample_colorscale(scale, samplepoints)
+    return result
+
+
+def resolve_colors(
+    colors: Colors,
+    px_kwargs: dict[str, Any],
+    *,
+    color_dim: Hashable | None = None,
+    darray: DataArray | Dataset | None = None,
+) -> dict[str, Any]:
     """Map unified `colors` parameter to appropriate Plotly px_kwargs.
 
     Direct color_* kwargs take precedence and trigger a warning if
@@ -260,6 +274,14 @@ def resolve_colors(colors: Colors, px_kwargs: dict[str, Any]) -> dict[str, Any]:
     Args:
         colors: Unified color specification (str, list, dict, or None).
         px_kwargs: Existing kwargs to pass to Plotly Express.
+        color_dim: Dimension name used for discrete color grouping.
+            When provided together with *darray*, a continuous colorscale
+            string is sampled into a discrete sequence whose length
+            matches the number of coordinates along this dimension.
+            Use for chart types that only accept discrete color
+            parameters (line, area, box, pie).
+        darray: Source DataArray or Dataset; used with *color_dim* to
+            determine the number of discrete colors to sample.
 
     Returns:
         Updated px_kwargs with color parameters injected.
@@ -284,6 +306,10 @@ def resolve_colors(colors: Colors, px_kwargs: dict[str, Any]) -> dict[str, Any]:
         # Check if it's a qualitative (discrete) palette name
         if colors in _get_qualitative_scale_names():
             px_kwargs["color_discrete_sequence"] = getattr(px.colors.qualitative, colors)
+        elif color_dim is not None and darray is not None:
+            # Sample from continuous scale into a discrete sequence
+            n = darray.sizes[color_dim]
+            px_kwargs["color_discrete_sequence"] = _sample_colorscale(colors, n)
         else:
             # Assume continuous scale
             px_kwargs["color_continuous_scale"] = colors

xarray_plotly/figures.py

@@ -308,6 +308,13 @@ def add_secondary_y(
     # Build mapping from primary y-axes to secondary y-axes
     y_mapping = _build_secondary_y_mapping(base_axes)
 
+    # Build x-y correspondence from base_axes (which x-axis pairs with which y-axis)
+    x_for_y = {yaxis: xaxis for xaxis, yaxis in base_axes}
+
+    # Find the rightmost x-axis (highest number) to determine which secondary axis shows ticks
+    rightmost_x = max(x_for_y.values(), key=lambda x: int(x[1:]) if x != "x" else 1)
+    rightmost_primary_y = next(y for y, x in x_for_y.items() if x == rightmost_x)
+
     # Create new figure with base's layout
     combined = go.Figure(layout=copy.deepcopy(base.layout))
 
@@ -322,24 +329,32 @@ def add_secondary_y(
         trace_copy.yaxis = y_mapping[original_yaxis]
         combined.add_trace(trace_copy)
 
+    # Get the rightmost secondary y-axis name for linking
+    rightmost_secondary_y = y_mapping[rightmost_primary_y]
+
     # Configure secondary y-axes
     for primary_yaxis, secondary_yaxis in y_mapping.items():
-        # Get title - only set on first secondary axis or use provided title
+        is_rightmost = primary_yaxis == rightmost_primary_y
+
+        # Get title - only set on rightmost secondary axis
         title = None
-        if secondary_y_title is not None:
-            # Only set title on the first secondary axis to avoid repetition
-            if primary_yaxis == "y":
+        if is_rightmost:
+            if secondary_y_title is not None:
                 title = secondary_y_title
-        elif primary_yaxis == "y" and secondary.layout.yaxis and secondary.layout.yaxis.title:
-            # Try to get from secondary's layout
-            title = secondary.layout.yaxis.title.text
+            elif secondary.layout.yaxis and secondary.layout.yaxis.title:
+                title = secondary.layout.yaxis.title.text
 
         # Configure the secondary axis
+        # Anchor to the corresponding x-axis so it appears on the right side of its subplot
         axis_config = {
             "title": title,
             "overlaying": primary_yaxis,
             "side": "right",
-            "anchor": "free" if primary_yaxis != "y" else None,
+            "anchor": x_for_y[primary_yaxis],
+            # Only show ticks on the rightmost secondary axis
+            "showticklabels": is_rightmost,
+            # Link non-rightmost axes to the rightmost for consistent scaling
+            "matches": None if is_rightmost else rightmost_secondary_y,
         }
         # Remove None values
         axis_config = {k: v for k, v in axis_config.items() if v is not None}

xarray_plotly/plotting.py

@@ -82,7 +82,6 @@ def line(
     -------
     plotly.graph_objects.Figure
     """
-    px_kwargs = resolve_colors(colors, px_kwargs)
     slots = assign_slots(
         list(darray.dims),
         "line",
@@ -94,6 +93,7 @@ def line(
         facet_row=facet_row,
         animation_frame=animation_frame,
     )
+    px_kwargs = resolve_colors(colors, px_kwargs, color_dim=slots.get("color"), darray=darray)
 
     df = to_dataframe(darray)
     value_col = get_value_col(darray)
@@ -329,7 +329,6 @@ def fast_bar(
     -------
     plotly.graph_objects.Figure
     """
-    px_kwargs = resolve_colors(colors, px_kwargs)
     slots = assign_slots(
         list(darray.dims),
         "fast_bar",
@@ -339,6 +338,7 @@ def fast_bar(
         facet_row=facet_row,
         animation_frame=animation_frame,
     )
+    px_kwargs = resolve_colors(colors, px_kwargs, color_dim=slots.get("color"), darray=darray)
 
     df = to_dataframe(darray)
     value_col = get_value_col(darray)
@@ -410,7 +410,6 @@ def area(
     -------
     plotly.graph_objects.Figure
     """
-    px_kwargs = resolve_colors(colors, px_kwargs)
     slots = assign_slots(
         list(darray.dims),
         "area",
@@ -421,6 +420,7 @@ def area(
         facet_row=facet_row,
         animation_frame=animation_frame,
     )
+    px_kwargs = resolve_colors(colors, px_kwargs, color_dim=slots.get("color"), darray=darray)
 
     df = to_dataframe(darray)
     value_col = get_value_col(darray)
@@ -487,7 +487,6 @@ def box(
     -------
     plotly.graph_objects.Figure
     """
-    px_kwargs = resolve_colors(colors, px_kwargs)
     slots = assign_slots(
         list(darray.dims),
         "box",
@@ -498,6 +497,7 @@ def box(
         facet_row=facet_row,
         animation_frame=animation_frame,
     )
+    px_kwargs = resolve_colors(colors, px_kwargs, color_dim=slots.get("color"), darray=darray)
 
     df = to_dataframe(darray)
     value_col = get_value_col(darray)
@@ -746,7 +746,6 @@ def pie(
     -------
     plotly.graph_objects.Figure
     """
-    px_kwargs = resolve_colors(colors, px_kwargs)
     slots = assign_slots(
         list(darray.dims),
         "pie",
@@ -754,6 +753,7 @@ def pie(
         facet_col=facet_col,
         facet_row=facet_row,
     )
+    px_kwargs = resolve_colors(colors, px_kwargs, color_dim=slots.get("names"), darray=darray)
 
     df = to_dataframe(darray)
     value_col = get_value_col(darray)

{xarray_plotly-0.0.12.dist-info → xarray_plotly-0.0.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xarray_plotly
-Version: 0.0.12
+Version: 0.0.14
 Summary: Interactive Plotly Express plotting accessor for xarray
 Author: Felix
 License: MIT
@@ -28,13 +28,13 @@ Provides-Extra: dev
 Requires-Dist: pytest==9.0.2; extra == "dev"
 Requires-Dist: pytest-cov==7.0.0; extra == "dev"
 Requires-Dist: mypy==1.19.1; extra == "dev"
-Requires-Dist: ruff==0.14.14; extra == "dev"
+Requires-Dist: ruff==0.15.5; extra == "dev"
 Requires-Dist: pre-commit==4.5.1; extra == "dev"
-Requires-Dist: nbstripout==0.9.0; extra == "dev"
+Requires-Dist: nbstripout==0.9.1; extra == "dev"
 Provides-Extra: docs
 Requires-Dist: mkdocs==1.6.1; extra == "docs"
-Requires-Dist: mkdocs-material==9.7.1; extra == "docs"
-Requires-Dist: mkdocstrings[python]==1.0.1; extra == "docs"
+Requires-Dist: mkdocs-material==9.7.4; extra == "docs"
+Requires-Dist: mkdocstrings[python]==1.0.3; extra == "docs"
 Requires-Dist: mkdocs-jupyter==0.25.1; extra == "docs"
 Requires-Dist: mkdocs-plotly-plugin==0.1.3; extra == "docs"
 Requires-Dist: jupyter==1.1.1; extra == "docs"
@@ -84,6 +84,25 @@ fig = xpx(da).line()
 
 Full documentation: [https://fbumann.github.io/xarray_plotly](https://fbumann.github.io/xarray_plotly)
 
+## Roadmap
+
+Planned additions (contributions welcome):
+
+**New plot types**
+- `histogram()` — distribution of DataArray values
+- `violin()` — richer distribution visualization than box plots
+- `density_heatmap()` — 2D histograms (x and y as dimensions)
+- `density_contour()` — 2D density contours (x and y as dimensions)
+
+**Enhancements**
+- WebGL rendering option for `line()` and `scatter()` (large datasets)
+
+**Figure utilities** (facet/animation-aware)
+- `fill_between()` — fill area between two traces (uncertainty bands)
+- `sync_axes()` — consistent axis ranges across facets and animation frames
+- `add_secondary_x()` — secondary x-axis (like `add_secondary_y()`)
+- `stack()` — vertically stack separate figures into subplots
+
 ## License
 
 MIT

xarray_plotly-0.0.14.dist-info/RECORD

@@ -0,0 +1,12 @@
+xarray_plotly/__init__.py,sha256=8YaSdbzAakzlRqkq9HkQlMxWOLGg2JfUN-gAkyCGf5U,3472
+xarray_plotly/accessor.py,sha256=XsGREdfGGqzFzpM53AzTbMnCOEtZTv0Sau-dc15meHQ,25313
+xarray_plotly/common.py,sha256=YdxqrK5LPTxDAALeMswxtTyEJuoHYUSess_28axJo2s,10782
+xarray_plotly/config.py,sha256=fOL7cqPHVj_XNLqN1oT2KQZ0DitIjzeYnWyMKcJy0cc,6679
+xarray_plotly/figures.py,sha256=uKC_8kcoFJYXsqMu-QRAsKfY8O_E3r8McBfRiHZxciE,16457
+xarray_plotly/plotting.py,sha256=OTz9ScASHnRQAJ9NvQAABiYALIoTrmYSLMEIZa4Zd44,24331
+xarray_plotly/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xarray_plotly-0.0.14.dist-info/licenses/LICENSE,sha256=AvVEfNqbhIm9jHvt0acJNjW1JUKa2a70Zb5rJdEXCJI,1064
+xarray_plotly-0.0.14.dist-info/METADATA,sha256=fKfz7Y0vuHRkhHfNGtObSxC9EWWKZr3eCYaCfzDyK_Y,4163
+xarray_plotly-0.0.14.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+xarray_plotly-0.0.14.dist-info/top_level.txt,sha256=GtMkvuZvLAYTjYXtwoNUa0ag42CJARZJK1CZemYD7pg,14
+xarray_plotly-0.0.14.dist-info/RECORD,,

{xarray_plotly-0.0.12.dist-info → xarray_plotly-0.0.14.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.2)
+Generator: setuptools (82.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

xarray_plotly-0.0.12.dist-info/RECORD

@@ -1,12 +0,0 @@
-xarray_plotly/__init__.py,sha256=8YaSdbzAakzlRqkq9HkQlMxWOLGg2JfUN-gAkyCGf5U,3472
-xarray_plotly/accessor.py,sha256=XsGREdfGGqzFzpM53AzTbMnCOEtZTv0Sau-dc15meHQ,25313
-xarray_plotly/common.py,sha256=XN5qDYG1UOGuocNekckujJNyDbWA9Q7pvdx8hC6w7OA,9576
-xarray_plotly/config.py,sha256=fOL7cqPHVj_XNLqN1oT2KQZ0DitIjzeYnWyMKcJy0cc,6679
-xarray_plotly/figures.py,sha256=kfTPkIpT0683XHBH-RuuM5cvuA_IzChDRcCSjOC0QEc,15709
-xarray_plotly/plotting.py,sha256=qHoEU1NX-YWkI23G60xYg_gzR_fEBSDcdrg0B2NpoPs,24106
-xarray_plotly/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xarray_plotly-0.0.12.dist-info/licenses/LICENSE,sha256=AvVEfNqbhIm9jHvt0acJNjW1JUKa2a70Zb5rJdEXCJI,1064
-xarray_plotly-0.0.12.dist-info/METADATA,sha256=3f-FDiVVKjJOt2GpbYC0mr0Q5F2tNPSNVPvWA7ejRRY,3416
-xarray_plotly-0.0.12.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-xarray_plotly-0.0.12.dist-info/top_level.txt,sha256=GtMkvuZvLAYTjYXtwoNUa0ag42CJARZJK1CZemYD7pg,14
-xarray_plotly-0.0.12.dist-info/RECORD,,