python-xarray - 沿 xarray 中的非坐标维度连接

Question

我正在尝试将一组值从一个DataArray连接到另一个。他们应该基于第一个的非维度坐标加入。我认为这应该很容易，但我似乎无法解决。

第一个数组：

In [4]: primary=xr.DataArray(np.random.rand(4), dims=list('a'))
   ...: primary.coords['group'] = (('a',), [0,0,1,1])
   ...: primary
   ...: 
Out[4]: 
<xarray.DataArray (a: 4)>
array([ 0.27772841,  0.06126117,  0.51753086,  0.35994987])
Coordinates:
  * a        (a) int64 0 1 2 3
    group    (a) int64 0 0 1 1

第二个数组：

In [5]: secondary=xr.DataArray(np.random.rand(2), dims=['group'], coords={'group': [0,1]})
   ...: secondary
   ...: 
Out[5]: 
<xarray.DataArray (group: 2)>
array([ 0.75944366,  0.90325489])
Coordinates:
  * group    (group) int64 0 1

我怎样才能primary - secondary使数组沿group维度连接？

谢谢

Answer 1

您需要groupby第一个但不对其应用聚合：

In [6]: primary.groupby('group') - secondary
Out[6]: 
<xarray.DataArray (a: 4)>
array([-0.48171525, -0.69818249, -0.38572403, -0.54330502])
Coordinates:
  * a        (a) int64 0 1 2 3
    group    (a) int64 0 0 1 1