如何移除Xarray的number维度并转为3D格式，通过rioxarray导出TIFF

问题分析 #

你遇到的核心问题是筛选number维度后出现了长度为0的空维度，常规的squeeze或drop_dims操作失效，导致导出时维度不符合rioxarray的要求。这大概率是因为筛选操作未正确匹配到数据，使得number维度变成空维度而非长度为1的维度。

分步解决方案 #

1. 先确保正确筛选number维度
   先确认number维度的坐标值，避免因索引不匹配导致空维度：

   # 查看number维度的所有取值
   print(ds.number.values)
   # 按坐标值筛选（如果number的坐标是[0,1]，选1）
   ds_selected = ds.sel(number=1)
   # 如果是按位置索引筛选（比如选第2个元素，索引从0开始）
   # ds_selected = ds.isel(number=1)
   

   此时ds_selected的number维度长度应为1，而非0。

2. 移除number维度
   针对长度为1的维度，使用squeeze并设置drop=True彻底移除，或直接用drop_dims：

   # 方法1：squeeze移除维度（drop=True会删除维度，而非仅压缩）
   ds_processed = ds_selected.squeeze(dim='number', drop=True)
   # 方法2：直接删除维度
   ds_processed = ds_selected.drop_dims('number')
   

3. 处理已出现的空维度（number:0）
   如果已经得到了空维度的数据集，先清理无效的空维度再移除：

   # 过滤掉number维度上全为空的条目
   ds_clean = ds.dropna(dim='number', how='all')
   # 再移除维度（此时clean后的维度长度应为1）
   ds_processed = ds_clean.squeeze(dim='number', drop=True)
   

4. 计算风速并导出TIFF
   处理完维度后计算风速，再用rioxarray导出（注意设置空间CRS）：

   # 假设u和v是风的水平分量，计算风速ws
   ds_processed['ws'] = (ds_processed['u'] ** 2 + ds_processed['v'] ** 2) ** 0.5
   
   # 设置空间坐标的CRS（根据你的数据实际CRS调整，示例为WGS84）
   ds_processed.rio.set_crs("EPSG:4326", inplace=True)
   
   # 导出TIFF，step维度会自动作为多波段
   ds_processed['ws'].rio.to_raster('wind_speed_output.tif')
   

关键注意点 #

• squeeze默认drop=False，只会压缩维度但不会删除，必须显式设置drop=True才能移除长度为1的维度。
• 若筛选后出现空维度，说明你的筛选条件没有匹配到任何数据，需要先检查number维度的实际取值是否正确。

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