python - 如何将数组中的零转换为Nan？

Question

我使用了temp[temp==0] = np.nan，但我得到了这个错误:

IndexError: 2-dimensional boolean indexing is not supported.

Answer 1

我会使用where，以避免必须下拉到numpy:

In [35]: d
Out[35]: 
<xarray.DataArray (dim_0: 2, dim_1: 3)>
array([[0, 1, 2],
       [3, 4, 5]])
Dimensions without coordinates: dim_0, dim_1

In [36]: d.where(d != 0)
Out[36]: 
<xarray.DataArray (dim_0: 2, dim_1: 3)>
array([[nan,  1.,  2.],
       [ 3.,  4.,  5.]])
Dimensions without coordinates: dim_0, dim_1

如有必要，它将自动移动到 float 。