Skip to content

Commit 723e19b

Browse files
committed
Squahs: multi-grouper propagate attrs
1 parent a0ab10f commit 723e19b

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

xarray/core/groupby.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -426,12 +426,16 @@ def factorize(self) -> EncodedGroups:
426426
)
427427
dim_name = "stacked_" + "_".join(str(grouper.name) for grouper in groupers)
428428

429+
coords = Coordinates.from_pandas_multiindex(midx, dim=dim_name)
430+
for grouper in groupers:
431+
coords.variables[grouper.name].attrs = grouper.group.attrs
432+
429433
return EncodedGroups(
430434
codes=first_codes.copy(data=_flatcodes),
431435
full_index=full_index,
432436
group_indices=_codes_to_group_indices(_flatcodes.ravel(), len(full_index)),
433437
unique_coord=Variable(dims=(dim_name,), data=midx.values),
434-
coords=Coordinates.from_pandas_multiindex(midx, dim=dim_name),
438+
coords=coords,
435439
)
436440

437441

0 commit comments

Comments
 (0)