when grouping by a flag array, set output coordinate using flag_meanings

[dcherian]

This would be cool and particularly useful with regionmask

cc @mathause

[mathause]

That would be super cool! Would need to decide if the flag_values are kept as well. I assume you would add this in da.cf.groupby(groups)?

A short example (using regionmask - could of course be simplified to only using xarray):

import cf_xarray
import regionmask
import xarray as xr

air = xr.tutorial.open_dataset("air_temperature")
mask = regionmask.defined_regions.ar6.land.mask(air)

regional_mean = air.air.groupby(mask).mean()

regional_mean = regional_mean.assign_coords({mask.name: mask.attrs["flag_values"]})
regional_mean = regional_mean.assign_coords({"flags": mask.attrs["flag_meanings"]})

# or
regional_mean = regional_mean.assign_coords({mask.name: mask.attrs["flag_meanings"]})