Add compose.Wrap

Author: has2k1

doc/_quartodoc.yml

@@ -540,6 +540,7 @@ quartodoc:
           include_attributes: false
         - Stack
         - Beside
+        - Wrap
         - plot_spacer
         - plot_layout
 

plotnine/_mpl/gridspec.py

@@ -27,8 +27,7 @@ class p9GridSpec(GridSpecBase):
     Gridspec for plotnine plots
 
     This gridspec does not read any subplot parameter values from matplotlib's
-    rcparams and the default a grid that fills up all the available space;
-    there is no space along the edges and between the subplots.
+    rcparams. And there is no space along the edges and between the subplots.
 
     This gridspec can also be initialised while contained/nested in a given
     subplot.

plotnine/_mpl/layout_manager/_grid.py

@@ -0,0 +1,94 @@
+from collections.abc import Iterator
+from dataclasses import InitVar, dataclass
+from typing import (
+    Generic,
+    Literal,
+    Sequence,
+    TypeVar,
+    overload,
+)
+
+import numpy as np
+
+T = TypeVar("T")
+
+
+@dataclass
+class Grid(Generic[T]):
+    nrow: InitVar[int]
+    ncol: InitVar[int]
+    items: InitVar[Sequence[T]]
+    order: InitVar[Literal["row_major", "col_major"]] = "row_major"
+    """
+    Put items into the grid left->right, top->bottom starting at `start`.
+    Unfilled cells remain None. Returns the number of items written.
+    """
+
+    def __post_init__(
+        self,
+        nrow: int,
+        ncol: int,
+        items: Sequence[T],
+        order: Literal["row_major", "col_major"],
+    ):
+        self._grid = np.empty((nrow, ncol), dtype=object)
+
+        r, c = 0, 0
+        if order == "row_major":
+            for item in items:
+                self[r, c] = item
+                c += 1
+                if c >= ncol:
+                    c, r = 0, r + 1
+        else:
+            for item in items:
+                self[r, c] = item
+                r += 1
+                if r >= nrow:
+                    c, r = r + 1, 0
+
+    @overload
+    def __getitem__(self, index: tuple[int, int]) -> T | None: ...
+
+    @overload
+    def __getitem__(self, index: tuple[int, slice]) -> list[T | None]: ...
+
+    @overload
+    def __getitem__(self, index: tuple[slice, int]) -> list[T | None]: ...
+
+    @overload
+    def __getitem__(
+        self,
+        index: tuple[slice, slice],
+    ) -> list[list[T | None]]: ...
+
+    def __getitem__(
+        self, index: tuple[int | slice, int | slice]
+    ) -> T | None | list[T | None] | list[list[T | None]]:
+        """
+        Access grid items with 2D indexing:
+        """
+        return self._grid[index]  # pyright: ignore[reportReturnType]
+
+    def __setitem__(
+        self,
+        index: tuple[int | slice, slice | int],
+        value: T | None | list[T | None] | list[list[T | None]],
+    ) -> None:
+        self._grid[index] = value
+
+    def iter_rows(self) -> Iterator[list[T | None]]:
+        """
+        Yield rows
+        """
+        n = self._grid.shape[0]
+        for row in range(n):
+            yield self[row, :]
+
+    def iter_cols(self) -> Iterator[list[T | None]]:
+        """
+        Yield cols
+        """
+        n = self._grid.shape[1]
+        for col in range(n):
+            yield self[:, col]