如何在 DataArray (xarray) 中组合两个坐标值？

Question

我有一个 xarray.Dataarray 有 4 个坐标： fp, station, run_date, elnu

尺寸目前（相同顺序）：（1、2、3261、417）

站有值“101470”和“108700”，想把这两个放在一起有一个维度（1，1，3261*2，417) 之后，我有点想重塑它们。问题是，我无法弄清楚如何做到这一点以及如何解决他们将拥有相同坐标的问题（如果我表现得像他们将拥有相同的车站）。

也许我只是在 google 上找不到合适的词来搜索（不是母语英语）。感谢任何帮助。

Answer 1

您正在寻找的功能在 xarray 的 .stack() 方法中。示例：

>>> import xarray

>>> import numpy as np

>>> da = xarray.DataArray(np.zeros((1, 2, 3261, 417)), dims=['fp', 'station', 'run_date', 'elnu'])

>>> da
<xarray.DataArray (fp: 1, station: 2, run_date: 3261, elnu: 417)>
array([[[[ 0., ...,  0.],
         ..., 
         [ 0., ...,  0.]],

        [[ 0., ...,  0.],
         ..., 
         [ 0., ...,  0.]]]])
Dimensions without coordinates: fp, station, run_date, elnu

>>> da.stack(station_date=['station', 'run_date'])
<xarray.DataArray (fp: 1, elnu: 417, station_date: 6522)>
array([[[ 0.,  0., ...,  0.,  0.],
        [ 0.,  0., ...,  0.,  0.],
        ..., 
        [ 0.,  0., ...,  0.,  0.],
        [ 0.,  0., ...,  0.,  0.]]])
Coordinates:
  * station_date  (station_date) MultiIndex
  - station       (station_date) int64 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...
  - run_date      (station_date) int64 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 ...
Dimensions without coordinates: fp, elnu

按照 .transpose() and/or .expand_dims() 将维度重新排序为您想要的顺序。

如何在 DataArray (xarray) 中组合两个坐标值？

How to combine two coordinate values in a DataArray (xarray)?

python

python-3.x

multidimensional-array

python-xarray