python - 沿 xarray 中的单个维度对多个坐标进行分组

Question

我有一个沿单个维度具有多个坐标的 xarray。在下面的例子中，坐标 a和 b沿尺寸 dim1 定义.我怎么会groupby使用沿相同维度定义的两个坐标？不像 this question ，我不是试图沿着不同的维度分组，而是一个单一的。

import xarray as xr

d = xr.DataArray([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]],
    coords={
        'a': ('dim1',['A', 'A', 'B', 'B']),
        'b': ('dim1',['1', '2', '1', '2']),
        'c': ('dim2',['x', 'y', 'z'])
    },
    dims=['dim1', 'dim2'])
d.groupby(['a','b']) # this gives: TypeError: `group` must be an xarray.DataArray or the name of an xarray variable or dimension

Answer 1

这是我目前的解决方法:

import numpy as np
import xarray as xr

def groupby_multicoords(da, fields):
    common_dim = da.coords[fields[0]].dims[0]
    tups_arr = np.empty(len(da[common_dim]), dtype=object)
    tups_arr[:] = list(zip(*(da[f].values for f in fields)))
    return da.assign_coords(grouping_zip=xr.DataArray(tups_arr, dims=common_dim)).groupby('grouping_zip')

然后， groupby_multicoords(da=d, fields=['a', 'b'])
但是，在分组之后，我仍然保留了“grouping_zip”坐标。我将不胜感激将其替换为 d.groupby(['a','b']) ..