如何将 xarray 数据集展平为一维 numpy 数组？

有没有一种简单的方法可以将 xarray 数据集展平为单个一维 numpy 数组？

例如，展平以下测试数据集：

xr.Dataset({
    'a' : xr.DataArray(
                   data=[10,11,12,13,14],
                   coords={'x':[0,1,2,3,4]},
                   dims={'x':5}
          ),
    'b' : xr.DataArray(data=1,coords={'y':0}),
    'c' : xr.DataArray(data=2,coords={'y':0}),
    'd' : xr.DataArray(data=3,coords={'y':0})
})

to

[10,11,12,13,14,1,2,3]

?

如果您同意重复值，则可以使用.to_array()然后将 NumPy 中的值展平，例如，

>>> ds.to_array().values.ravel()
array([10, 11, 12, 13, 14,  1,  1,  1,  1,  1,  2,  2,  2,  2,  2,  3,  3,
        3,  3,  3])

如果你不想重复的值，那么你需要自己写一些东西，例如，

>>> np.concatenate([v.values.ravel() for v in ds.data_vars.values()])
array([10, 11, 12, 13, 14,  1,  2,  3])

更一般地说，这听起来有点类似于为机器学习应用程序“堆叠”二维数据变量的提议接口：https://github.com/pydata/xarray/issues/1317 https://github.com/pydata/xarray/issues/1317