如何在xarray中移除含nan变量对应的坐标及数组

文件情况说明 #

• 原始NetCDF文件：data1.nc，可视化效果：
  • 
  • 
• 掩膜处理后文件：new.nc（已将海洋数值转为nan），可视化效果：
  • 
  • 

需求目标 #

从new.nc中移除变量r和t对应的含nan的坐标，同时删除这些坐标对应的所有NaN数组。

实操方案（Python + xarray） #

1. 安装依赖（未安装时执行） #

pip install xarray netcdf4

2. 核心处理代码 #

import xarray as xr
import numpy as np

# 读取目标文件
ds = xr.open_dataset("new.nc")

# 筛选出r、t变量中不含NaN的坐标索引
valid_r = ~np.isnan(ds['r'])
valid_t = ~np.isnan(ds['t'])

# 保留有效坐标及对应的数据
ds_clean = ds.sel(r=valid_r, t=valid_t)

# 保存处理后的文件
ds_clean.to_netcdf("cleaned_new.nc")

3. 特殊情况处理 #

如果r和t不是维度坐标，而是普通数据变量，需要先定位它们关联的维度再筛选：

# 示例：假设r关联维度x，t关联维度y，需根据实际数据结构调整
# 找出x维度中r无NaN的索引
valid_x = ~np.isnan(ds['r']).any(dim=['time'])
# 找出y维度中t无NaN的索引
valid_y = ~np.isnan(ds['t']).any(dim=['time'])
# 筛选数据集
ds_clean = ds.sel(x=valid_x, y=valid_y)

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