8

有没有一种简单的方法可以将 xarray 数据集展平为单个 1D 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})
})

至

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

?

4

3 回答 3

6

如果您对重复值没问题，您可以使用.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])

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

于 2017-10-31T18:14:13.053 回答
6

截至 2019 年 7 月，xarray 现在具有执行此功能的函数to_stacked_array和to_unstacked_dataset 。

于 2019-07-11T10:27:10.537 回答
1

从问题中获取数据集：

ds = 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})
})

获取数据变量列表：

variables = ds.data_vars

使用该np.flatten() 方法将数组减少为一维：

arrays = [ ds[i].values.flatten() for i in variables ]

然后展开一维数组列表（如本答案中所述）：

arrays = [i for j in arrays for i in j  ]

现在将其转换为 Q 中请求的数组（当前为列表）：

array = np.array(arrays)
于 2017-11-13T10:42:37.533 回答