为何赋值xr.ones_like给Dataset变量会改变无关坐标并丢失数据？

我来帮你拆解这个xarray使用时遇到的坐标问题，先把完整的复现代码补全，方便复现现象：

import xarray as xr
import numpy as np

A, B, C = 2, 3, 4
ds = xr.Dataset()
# 定义source坐标，维度为a,b,c
ds.coords['source'] = (['a', 'b', 'c'], np.random.random((A, B, C)))
# 定义unrelated坐标，维度为a,c
ds.coords['unrelated'] = (['a', 'c'], np.random.random((A, C)))

print('INITIAL:', ds['unrelated'], '\n')

# 手动创建全1数组赋值，无异常
ds['dest-1'] = (['a', 'b'], np.ones((A, B)))
print('AFTER dest-1:', ds['unrelated'], '\n')

# 使用xr.ones_like赋值，触发异常
ds['dest-2'] = xr.ones_like(ds['source'])
print('AFTER dest-2:', ds['unrelated'], '\n')

问题原因 #

• xr.ones_like的特性：它会完全复刻原变量（这里是ds['source']）的所有维度、坐标属性，包括c维度的坐标信息。
• xarray的自动坐标对齐机制：当你把带有c维度的DataArray赋值给Dataset时，xarray会尝试对齐Dataset中所有相关的坐标。但unrelated变量本身就有c维度，赋值过程中会错误地将dest-2的c坐标覆盖到unrelated上，甚至因为维度匹配问题导致部分坐标数据丢失。
• 手动赋值的差异：手动指定维度['a','b']时，赋值的数组不携带任何额外坐标，不会触发xarray的自动坐标对齐，因此不会影响unrelated坐标。

解决方案 #

要避免这个问题，核心是确保赋值的DataArray只携带目标维度的坐标，不要引入无关维度，以下是几种可行的方法：

方法1：直接指定维度创建全1数组（最简洁）

# 只保留需要的a、b维度，不携带多余坐标
ds['dest-2'] = xr.DataArray(np.ones((A, B)), dims=['a', 'b'])

方法2：从原变量提取目标维度后创建ones_like

如果需要基于原变量的维度信息，可以先提取目标维度的子集，再创建ones_like：

# 提取source的a、b维度（去掉c维度），再创建ones_like
source_ab = ds['source'].isel(c=0).drop_vars('c')
ds['dest-2'] = xr.ones_like(source_ab)

方法3：用xr.ones显式指定维度和对应坐标

如果需要保留目标维度的坐标信息，可以直接用xr.ones指定维度和坐标：

ds['dest-2'] = xr.ones(
    dims=['a', 'b'],
    coords={'a': ds.coords['a'], 'b': ds.coords['b']}
)

验证结果 #

使用上述任意方法赋值后，再打印ds['unrelated']，会发现它的维度和数据都保持初始状态，不会被修改或丢失。

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