如何在Dask后端的DataArray沿现有维度原地追加新数据？

方法一：基于DataArray直接处理 #

1. 正确计算日收益率并设置坐标 #

先提取收盘价数据，计算日收益率后，为结果添加正确的indicators维度和坐标，避免合并时出现重复：

# 提取收盘价，drop=True得到(time, coins)二维数组
price = da.sel(indicators="price_daily_close_usd", drop=True)
# 计算日收益率（假设昨日数据不为0）
daily_return = price / price.shift(time=1) - 1
# 将二维数组扩展为三维，指定新指标的坐标
daily_return_da = daily_return.expand_dims(indicators=["daily_return"]).assign_coords(indicators=["daily_return"])

2. 合并原数据与新指标 #

沿indicators维度拼接原数组和新生成的日收益率数组：

new_da = xr.concat([da, daily_return_da], dim="indicators")

方法二：切换为Dataset结构（更贴近DataFrame新增列的逻辑） #

Dataset的变量操作更直观，类似pandas添加列的方式：

# 将DataArray转换为Dataset，每个指标对应一个变量
ds = da.to_dataset(dim="indicators")
# 计算日收益率并直接添加为新变量
ds["daily_return"] = ds["price_daily_close_usd"] / ds["price_daily_close_usd"].shift(time=1) - 1
# 若需转回DataArray格式
new_da = ds.to_array(dim="indicators")

关键说明 #

1. 关于原地修改：xarray和Dask均为不可变数据结构，无法实现原地修改。但生成新对象仅构建计算任务图，不会立即加载数据到内存，完全适配大数据量场景。
2. 解决之前的坐标问题：之前的错误源于未给新指标设置独立的indicators坐标，只要确保新生成的日收益率数组的indicators维度值为daily_return，合并后就不会出现重复坐标或索引错误。

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