risky

Author: hmaarrfk

xarray/core/indexes.py

@@ -1480,12 +1480,14 @@ def _id_index(self) -> dict[int, T_PandasOrXarrayIndex]:
     @property
     def _id_coord_names(self) -> dict[int, tuple[Hashable, ...]]:
         if self.__id_coord_names is None:
-            id_coord_names: Mapping[int, list[Hashable]] = defaultdict(list)
-            for k, v in self._coord_name_id.items():
-                id_coord_names[v].append(k)
-            self.__id_coord_names = {k: tuple(v) for k, v in id_coord_names.items()}
-
+            self._create_id_coord_names()
         return self.__id_coord_names
+    
+    def _create_id_coord_names(self) -> None:
+        id_coord_names: Mapping[int, list[Hashable]] = defaultdict(list)
+        for k, v in self._coord_name_id.items():
+            id_coord_names[v].append(k)
+        self.__id_coord_names = {k: tuple(v) for k, v in id_coord_names.items()}
 
     @property
     def variables(self) -> Mapping[Hashable, Variable]:
@@ -1771,6 +1773,25 @@ def check_variables():
 
     return not not_equal
 
+def _apply_indexes_isel(
+    indexes: Indexes[Index],
+    args: Mapping[Any, Any],
+):
+    new_indexes: dict[Hashable, Index] = {k: v for k, v in indexes.items()}
+    new_index_variables: dict[Hashable, Variable] = {}
+#     for index, index_vars in indexes.group_by_index():
+    for index in indexes._id_index.values():
+        # index = indexes._id_index[i]
+        index_dims = (index.index.name,)
+        index_args = {k: v for k, v in args.items() if k in index_dims}
+        if index_args:
+            new_index = index.isel(index_args)
+            if new_index is not None:
+                new_indexes.update({k: new_index for k in index_vars})
+                new_index_vars = new_index.create_variables(index_vars)
+                new_index_variables.update(new_index_vars)
+
+    return new_indexes, new_index_variables 
 
 def _apply_indexes(
     indexes: Indexes[Index],
@@ -1800,7 +1821,7 @@ def isel_indexes(
     indexes: Indexes[Index],
     indexers: Mapping[Any, Any],
 ) -> tuple[dict[Hashable, Index], dict[Hashable, Variable]]:
-    return _apply_indexes(indexes, indexers, "isel")
+    return _apply_indexes_isel(indexes, indexers)
 
 
 def roll_indexes(