xarray where函数

xarray 的 where 函数是一个条件语句，它可以返回一个布尔值的掩码数组，用于选择符合特定条件的数据。其语法如下：

xarray.where(condition, other=nan, drop=True)

参数说明：

- `condition`：条件语句，可以是一个布尔数组、xarray.DataArray 或标量值，必需。
- `other`：可选参数，当条件不满足时，要用来替代数据的值，默认为 NaN。
- `drop`：可选参数，是否删除不满足条件的维度，默认为 True。

举个例子，假设我们有一个数据集 `ds`，其中包含 `temperature` 和 `precipitation` 两个变量，并且我们想要选择温度大于 25 度的数据，可以使用如下代码：

import xarray as xr

ds = xr.open_dataset('data.nc')
temp_gt_25 = ds.where(ds.temperature > 25)

这将返回一个新的数据集 `temp_gt_25`，其中只包含温度大于 25 度的数据，其他数据将被替换为 NaN。