如何替换xarray.DataArray的子集值？替换操作无生效问题解析

你的问题出在：Xarray中，通过链式索引（比如ds.isel(...)[var]）得到的是原数据的副本或临时视图，对这些临时对象赋值不会修改原始数据集。下面是几种有效的修改方式：

方法1：直接对变量进行索引赋值 #

直接定位到原数据集的目标变量，再通过索引选择子集进行赋值，这是最直接的方式：

# 用位置索引（isel）
ds['tmin'].isel(location=0) = np.arange(ds.dims['time'])

# 或者用标签索引（loc），更直观
ds['tmin'].loc[{'location': 'IA'}] = np.arange(ds.dims['time'])

方法2：创建新变量后替换原数据集 #

如果需要保留原数据集不变，生成新的数据集，可以用assign方法：

# 复制原tmin数组
new_tmin = ds['tmin'].copy()
# 修改子集
new_tmin.isel(location=0) = np.arange(ds.dims['time'])
# 替换原数据集的tmin变量
ds = ds.assign(tmin=new_tmin)

方法3：使用xr.where条件赋值 #

如果需要基于条件批量修改，xr.where是更灵活的选择：

ds['tmin'] = xr.where(
    ds.location == 'IA',  # 条件：location为IA
    np.arange(ds.dims['time']),  # 满足条件时的新值
    ds['tmin']  # 不满足条件时保留原值
)

为什么你的原有方法不生效？ #

• 第一种方法ds.isel(location=0).tmin.values = ...：ds.isel(location=0)返回的是原数据集的临时子集，修改这个子集的.values只会改变临时对象的数组，不会同步到原数据集。
• 第二种方法ds.isel(location=0)['tmin'] = da_new：ds.isel(location=0)生成的是一个新的Dataset副本，对副本的变量赋值不会影响原始数据集。

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