xarray乘法运算后数组的形状不正确

Question

根据Value error in multplying xarray variable with 2D numpy array

import xarray as xr
hndl_tran = xr.open_dataset(path_netcdf, chunks={'time': 10})
flow_data = hndl_tran['val']
new_arr = flow_data * xr.DataArray(vba)

以下是输入数组的形状

flow_data.shape
(1165, 720, 1440)

vba.shape
(720L, 1440L)

以下是乘以后数组的形状：

new_arr.shape
(1165, 720, 1440, 720, 1440)

我希望生成的数组与flow_data具有相同的形状。我该怎么做？

Answer 1

xarray根据数组的尺寸对齐形状。因此，如果维度不共享名称，则乘法将创建所有维度的并集。

我认为flow_data和vba具有不同的命名维度 - 使用.rename将匹配的维度设置为匹配的名称