Refactoring GroupBy

1. Simplify ResolvedGrouper by moving logic to EncodedGroups 2. Stack outside ResolvedGrouper in GroupBy.__init__ to prepare for multi-variable GroupBy
pydata · Aug 21, 2024 · 5b40ae3 · 5b40ae3
1 parent 3107484
commit 5b40ae3
Show file tree

Hide file tree

Showing 4 changed files with 200 additions and 162 deletions.
diff --git a/xarray/core/coordinates.py b/xarray/core/coordinates.py
@@ -352,7 +352,7 @@ def _construct_direct(
         return obj
 
     @classmethod
-    def from_pandas_multiindex(cls, midx: pd.MultiIndex, dim: str) -> Self:
+    def from_pandas_multiindex(cls, midx: pd.MultiIndex, dim: Hashable) -> Self:
         """Wrap a pandas multi-index as Xarray coordinates (dimension + levels).
 
         The returned coordinates can be directly assigned to a