Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Ordered Groupby Keys

See original GitHub issue

The current behavior of the xarray’s Groupby.groups property provides a standard (unordered) dictionary. This is fine for most cases but leads to odd orderings in use cases like this one where I am using xarray’s FacetGrid plotting:

plot_kwargs = dict(col='season', vmin=15, vmax=35, levels=12, extend='both')

da_obs = ds_obs.SALT.isel(depth=0).groupby('time.season').mean('time')
da_obs.plot(**plot_kwargs)

index Note that MAM and JJA are out of order.

I think this could be easily fixed by using an OrderedDict in xarray.core.Groupby.groups.

Issue Analytics

State:
Created 8 years ago
Comments:6 (5 by maintainers)

Top GitHub Comments

2reactions

dcheriancommented, Nov 20, 2019

I use reindex instead of sortby

0reactions

jbuseckecommented, Nov 20, 2019

Just stumbled across this issue. Is there a recommended workaround?

I am usually doing this (specific to seasons):

import xarray as xr
ds = xr.tutorial.open_dataset('air_temperature')
airtemp_seasonal = ds.groupby('time.season').mean('time').sortby(xr.DataArray(['DJF','MAM','JJA', 'SON'],dims=['season']))

Thought this might help some folks who need to solve this problem.