Skip to content

Commit 914a1a1

Browse files
committed
fix assign_coords multi-index: update coord names
1 parent e678a1d commit 914a1a1

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

xarray/core/coordinates.py

+1
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,7 @@ def _maybe_drop_multiindex_coords(self, coords: set[Hashable]) -> None:
315315
variables, indexes = drop_coords(
316316
coords, self._data._variables, self._data.xindexes
317317
)
318+
self._data._coord_names.intersection_update(variables)
318319
self._data._variables = variables
319320
self._data._indexes = indexes
320321

xarray/tests/test_dataset.py

+4
Original file line numberDiff line numberDiff line change
@@ -4146,6 +4146,10 @@ def test_assign_coords_existing_multiindex(self) -> None:
41464146
):
41474147
data.assign(x=range(4))
41484148

4149+
# https://github.com/pydata/xarray/issues/7097 (coord names updated)
4150+
updated = data.assign_coords(x=range(4))
4151+
assert len(updated.coords) == 1
4152+
41494153
def test_assign_all_multiindex_coords(self) -> None:
41504154
data = create_test_multiindex()
41514155
actual = data.assign(x=range(4), level_1=range(4), level_2=range(4))

0 commit comments

Comments
 (0)