0

我正在使用xarray's groupby+ reducer 对空间栅格执行空间叠加/聚合。我想知道是否有办法为某些数据变量使用不同的减速器。例如，在下面的代码中，我想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()
4

2 回答 2

2

目前在目前 AFAIK 的 xarray 中这是不可能一次性完成的，但是由于无论如何你都失去了空间结构，你可以非常简单地通过 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
于 2018-03-20T15:08:51.567 回答
0

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

xr_dataset = xr.merge([
  xr_dataset.categorical_variable.groupby('grouping_variable').first(),
  xr_dataset.continuous_variable.groupby('grouping_variable').mean(),
  ...
])
于 2022-02-25T17:24:23.037 回答