沿第三维查找 xarray DataArray 的最高值。

Question

我有一个 3 维 xarray DataArray，其中两个维度表示一个 xy 网格，第三个维度表示彼此“堆叠”的网格层数。 空的 DataArray 看起来像这样：

import xarray as xr
import numpy as np

data = np.zeros(shape=(layers,y,x))
dims=['layer','y_Axis', 'x_Axsis']
dataArray = xr.DataArray(data,dims=dims)

在一个例程中，我用 -1 和 1 之间的值逐层填充网格。

我现在的任务是将所有层堆叠在一起，选择所有网格组合的最高值。 因此，当例如 5 层相互比较时，网格内的最高元素将保存在 2d numpy 数组中。

我可以通过遍历每一层来解决这个问题，将当前层的网格值与创建的 2d max_val_grid 进行比较，从而过滤第三维。 然而，这听起来非常低效。

你们有没有人知道如何使用内部 xarray 或 numpy 函数而不循环来解决这个问题？

多谢！

Answer 1

您可以使用numpy.amax如下

numpy.amax(your_3D_array, axis=2)

这将在 3D 数据的最后一个轴上选择your_3D_array的最大值并返回一个 2D 数组。 这是一个快速测试用例：

import numpy as np

x = np.arange(10)
y = np.arange(10, 20)
z = np.arange(20, 30)

x, y, z = np.meshgrid(x, y, z)

print(np.amax(z, axis=2))

Answer 2

xarray 中的max()方法完全符合您的要求，例如dataArray.max('layer') 。