我們如何塑造 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) 形狀。