Xarray切片函数替代沿维度计算平均值的替代方案

Question

我正在使用Xarray和NetCDF气象数据。我有通常的尺寸时间，纬度和经度以及两个主要变量：风速（时间，lat，lon）和纬度位置（时间，lon）。

 <xarray.Dataset>
Dimensions:             (lon: 53, time: 25873, lat: 20)
Coordinates:
  * lon                 (lon) float64 -80.0 -77.5 -75.0 -72.5 ... 45.0 47.5 50.0
  * time                (time) datetime64[ns] 1950-01-31 ... 2020-12-01
  * lat                 (lat) float32 70.0 67.5 65.0 62.5 ... 27.5 25.0 22.5
Data variables:
    uwnd                (time, lat, lon) float32 -0.0625 0.375 ... -1.812 -2.75
    positions           (time, lon) float64 40.0 40.0 45.0 ... 70.0 70.0 70.0

每次，LON，我都想计算位置周围的纬度平均值。

如果我做一个循环，我会这样做（对于 +-2.5°平均平均水平）：

for i in ds.lon.values:
        for t in ds.time.values:
              wind_averaged.loc[t,i]=ds.uwnd.sel(lon=i,time=t).sel(lat=slice(2.5+ds.positions.sel(lon=i,time=t).values,ds.positions.sel(lon=i,time=t).values-2.5)).mean('lat')

这显然很糟糕，我想使用 slice（）这样：

wind_averaged=ds.uwnd.sel(lat=slice(2.5+ds.jet_positions.values,ds.jet_positions.values-2.5)).mean('lat')

但是它给出了错误，因为我

cannot use non-scalar arrays in a slice for xarray indexing

有其他选择可以通过使用Xarray Power做我想做的事情吗？

谢谢

Answer 1

我相信您正在寻找多维集团。如果我正确理解的话，这里有一个有关此问题的教程：