python - 将 xarray 变量与 2D numpy 数组相乘时的值错误

Question

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

我收到此错误：

*** ValueError: total size of new array must be unchanged

以下是 2 个数组的形状：

flow_data.shape
(1165, 720, 1440)

vba.shape
(720L, 1440L)

我该如何解决这个错误？

Answer 1

xarray在进行乘法之前，将您的 numpy 变成一个对象：

flow_data = xr.DataArray(hndl_tran['val'])

或相反亦然

flow_data = np.array(flow_data)

Answer 2

在@maxymoo 的答案的基础上，您希望转换为 a DataArray，但也提供 dims，因此与其他数组的操作将起作用flow_data = xr.DataArray(hndl_tran['val'], dims=['date', 'id'])，将 date & id 替换为适当的名称