如何更改dimensions xarray.dataset中的顺序？

Question

我正在创建一个如下：

import numpy as np
import xarray as xr

x_example = np.random.rand(1488,)
y_example = np.random.rand(1331,)
time_example = np.random.rand(120,)
rainfall_example = np.random.rand(120, 1331, 1488)

rainfall_dataset = xr.Dataset(
    data_vars=dict(
        rainfall_depth=(['time', 'y', 'x'], rainfall_example),
    ),
    coords=dict(
        time=(['time'], time_example),
        x=(['x'], x_example),
        y=(['y'], y_example)
    )
)

结果类似于

和我运行时的尺寸rainfall_example.dims就像冷冻（{'time'：120，'y'：1331，'x'：1488}）（也可以在上述结果）。我知道xarray.dataset.dims不能根据在这里

我的问题是：我们如何将这些维度的顺序更改为这样的维度冷冻（{'time'：120，'x'：1488，'y'：1331}）而无需更改其他任何内容（一切都将相同，只有更改维度的顺序）？

Answer 1

您可以通过选择两个列表来选择两个坐标和变量来重新排序坐标和变量：

In [3]: rainfall_dataset[["time", "y", "x", "rainfall_depth"]]
Out[3]:
<xarray.Dataset>
Dimensions:         (time: 120, y: 1331, x: 1488)
Coordinates:
  * time            (time) float64 0.2848 0.7556 0.9501 ... 0.694 0.734 0.198
  * y               (y) float64 0.1941 0.1132 0.2504 ... 0.1501 0.5085 0.006135
  * x               (x) float64 0.2776 0.4504 0.1886 ... 0.4071 0.3327 0.5555
Data variables:
    rainfall_depth  (time, y, x) float64 ...

Answer 2

如何更改Xarray的维度顺序：

在您的示例中，那将是：

rainfall_dataset['rainfall_depth'].T.transpose('time','x','y')

IE：

Dataset['Variable'].T.transpose('dim1','dim2','dim3)

必需：列出所需的所有维度订购。

如果您想在数据集中修改变量：

rainfall_dataset['rainfall_depth'] = rainfall_dataset['rainfall_depth'].T.transpose('time','x','y')

IE：

Dataset['Variable'] = Dataset['Variable'].T.transpose('dim1','dim2','dim3)