如何在xarray DataArray数值运算时保留NaN并排除其参与计算

你的原始DataArray是uint16无符号整数类型，该类型本身不支持存储NaN空值。属性里标记的_FillValue=0.0只是元数据声明，xarray执行运算时不会自动把值为0的填充位识别为空值，所有0值会被当作有效像素参与计算，最终导致原本的空值位置全部变成固定运算结果。

先将填充位转换为真正可参与运算的NaN值，再执行常数乘法即可。NaN在算术运算中会自动传播，始终保留为空值，不会被替换为计算结果，两种常用实现方式如下：

• 手动识别填充值转NaN
  直接通过where方法过滤填充值，逻辑直观可控：

  # 假设原始DataArray变量名为da，待乘常数为target_constant
  # 把等于填充值的位置设为NaN，其余值保留，再执行乘法
  result = da.where(da != da.attrs["_FillValue"]) * target_constant
  

• 自动解码CF规范元数据
  如果你的数据是读入的遥感栅格等符合CF规范的产品，可以直接调用xarray内置解码函数，自动识别_FillValue、scale_factor、add_offset等属性完成预处理：

  import xarray as xr
  # 自动解码元数据，将填充位替换为NaN
  da_decoded = xr.decode_cf(da)
  # 解码后直接执行运算即可，空值位置会自动保留为NaN
  result = da_decoded * target_constant
  

  如果你是通过rioxarray.open_rasterio读取栅格数据，读取时直接传入参数mask_and_scale=True，在读入阶段就会自动完成填充值识别转NaN的操作，后续运算无需额外预处理。

• 转换为NaN后数据类型会自动切换为浮点型（整数类型无法存储NaN），属于正常现象。
• 如果后续需要将结果导出为tif等栅格文件，可在导出前通过result = result.fillna(0)将NaN重新映射回0值作为填充值，同步更新_FillValue属性即可。

内容的提问来源于stack exchange，提问作者MaxDragonheart