xarray groupby：对变量应用不同的缩减器

Question

我正在使用 xarray 的 groupby + reducer 在空间栅格上执行空间 overlay/aggregation。我想知道是否有办法对某些数据变量使用不同的减速器。例如，在下面的代码中，我希望 categorical_variable 减少为 first() （或 mode 但似乎没有实现），并且 continuous_variable 是减少 mean()

import xarray as xr
import numpy as np

categorical_variable = np.array([[1,1,1,1,1],
                                 [1,1,1,1,2],
                                 [1,1,1,2,2],
                                 [1,1,2,2,2],
                                 [1,2,2,2,2]], dtype='int16')
grouping_variable = np.array([[1,1,1,2,2],
                              [1,1,3,2,2],
                              [1,3,3,3,3],
                              [3,3,3,3,3],
                              [4,4,4,4,4]], dtype='int16')
continuous_variable = np.random.rand(5,5)
xr_dataset = xr.Dataset({'grouping_variable': xr.DataArray(grouping_variable,
                                                           dims=['x', 'y']),
                         'categorical_variable': xr.DataArray(categorical_variable,
                                                              dims=['x', 'y']),
                         'continuous_variable': xr.DataArray(continuous_variable,
                                                             dims=['x', 'y'])})
xr_grouped = xr_dataset.groupby('grouping_variable')
xr_reduced = xr_grouped.mean()

Answer 1

目前在 xarray 中目前不可能一次完成 AFAIK，但是由于您无论如何都会丢失空间结构，因此您可以非常简单地通过 pandas 并使用 agg:

>>> df = xr_dataset.to_dataframe()
>>> df.groupby('grouping_variable').agg({"categorical_variable": "first",
                                         "continuous_variable": "mean"})

                   categorical_variable  continuous_variable
grouping_variable                                           
1                                     1             0.458534
2                                     1             0.822294
3                                     1             0.539483
4                                     1             0.515586

Answer 2

性能不是最佳的，但这是我最终做的：

xr_dataset = xr.merge([
  xr_dataset.categorical_variable.groupby('grouping_variable').first(),
  xr_dataset.continuous_variable.groupby('grouping_variable').mean(),
  ...
])

xarray groupby：对变量应用不同的缩减器

xarray groupby: Apply different reducers to variables

python

numpy

python-xarray

xarray