python-3.x - 屏蔽 xarray 中的数据并更改 True 和 False 响应的值

Question

我有一个 xarray DataArray，其中包含多天的数据。我可以在一个条件下使用 .where 函数对其进行屏蔽，但我想将某个值以上的所有值设为 1，并将该值以下的所有值设为 0。理想情况下，我还想确保任何 np.数据集中的 nans 没有改变，但这不是必需的。

import numpy as np
import xarray as xr

dval = np.random.randint(5,size=[3,4,4])
x = [0,1,2,3]
y = [0,1,2,3]
time = ['2017-10-13','2017-10-12','2017-10-11']

a = xr.DataArray(dval,coords=[time,x,y],dims=['time','x','y'])
a = a.where(a>2,1,0) #ideally this would work as (condition,True val, False val)

这会导致 ValueError “如果 drop=True，则无法设置‘其他’”

对此的任何帮助将不胜感激。

Answer 1

a = a.where(a>2, 1, 0)不起作用，因为该DataArray.where方法仅支持设置other。基本上你在做：a = a.where(a>2, other=1, drop=0).

相反，您应该使用 xarray 的 3 参数xr.where函数：

a= xr.where(a>2, 1, 0)