python - 创建一个 xarray.DataArray，其中所有值都满足条件

Question

我有一个一维的 DataArray，每个值对应一个 ID。我正在尝试以相同的格式创建一个新数组，其中包含值超出限制的任何情况下的值和 ID

我尝试过使用xr.DataArray.where(condition,other)，但是对于不满足条件的情况，这会生成一个与“nan”大小相同的数组。

#create mock array
ID = np.arange(10)
values = np.random.uniform(0.5, 20, 10)
xr.DataArray(values, dims='ID', coords={'ID':ID})

xlim = 10

示例输出可以是一个包含 5 个元素的数组(所有元素的值均大于 10，即我们设置的限制)以及原始数组中相应的 ID。

Answer 1

您可以直接使用 numty 索引。

#create mock array
ID = np.arange(10)
values = np.random.uniform(0.5, 20, 10)
arr = xr.DataArray(values, dims='ID', coords={'ID':ID})

xlim = 10

arr[np.where(arr>xlim)]

<xarray.DataArray (ID: 4)> array([18.830967, 14.651416, 17.545492,
17.627902]) Coordinates:   * ID       (ID) int64 2 7 8 9