子集化xarray.DataArray时保留Dataset中的自定义坐标

核心问题分析 #

你遇到的问题本质是未正确创建非维度坐标（即依附于现有维度的坐标，而非新增独立维度）。错误操作导致生成了不必要的year维度，且子集化时丢失关联坐标。

正确添加多套坐标系的方法 #

要为time维度同时保留datetime坐标和数值型year/季节坐标，需确保新坐标与time维度长度一致，作为time的附属坐标而非新维度：

1. 为Dataset添加附属坐标

直接利用Xarray的时间属性提取年份/季节，通过assign_coords绑定到time维度：

import xarray as xr

# 假设ds是你的NASA数据集对象
ds = ds.assign_coords(
    year=ds.time.dt.year,
    season=ds.time.dt.season  # 可选：添加季节坐标
)

验证坐标关联：执行ds.coords['year'].dims，输出应为('time',)，说明year是time的附属坐标，而非独立维度。

2. 确保子集化时保留坐标

从Dataset提取DataArray并子集化后，关联坐标会自动保留：

# 提取目标DataArray
da = ds['your_variable_name']

# 子集化操作（示例：筛选2000-2010年数据）
da_subset = da.sel(time=slice('2000-01-01', '2010-12-31'))

# 验证坐标是否存在
print(da_subset.coords)  # 应包含time、year、season（若添加）

3. 解决ValueError: cannot add coordinates with new dimensions错误

该错误源于添加的坐标长度与目标维度不匹配，或误将坐标作为新维度添加。给单个DataArray添加坐标时，需确保坐标长度与time维度完全对齐：

# 为单个DataArray添加year坐标
da = da.assign_coords(year=da.time.dt.year)

关键注意事项 #

• 区分维度坐标与非维度坐标：维度坐标是维度的索引（如默认的datetime64类型time坐标），非维度坐标依附于现有维度，用于补充信息（如每个time对应的year）。
• 避免手动创建长度不匹配的坐标数组，直接利用dt属性提取时间信息能确保坐标与time维度完全对齐。

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