我们如何塑造 Xarray 数据数组？

Question

我一直在使用 x-array。 我们如何重塑数组？ 在 NumPy 中是否有任何类似 re-shape 的方法？

temp = data['__xarray_dataarray_variable__'][:, :, :]

上面的代码行选择了 x 数组的所有 3 个维度。 即数组的形状是(5, 2, 1)。 如何更改 x 数组，使其具有 (5*2, 1) 的形式。 提前致谢。

Answer 1

Xarray 语法有点不直观。 在任何情况下，您都可以使用堆栈：

temp = temp.stack(dim_new=['dim_0','dim_1'])

假设最初它是一个形状为(dim_0: 5, dim_1: 2, dim_2: 1)的xarray，那么 temp 将成为一个尺寸为(dim_2: 1, dim_new: 10)的xarray。 您可以进一步应用temp.transpose('dim_new','dim_2')使其变为 (10,1) 形状。