具有二级索引的一维 xarray

Question

我想要一个带有 "primary" 索引的一维数组，然后将其他索引添加到同一轴。例如，如果我的数据是这样的，

[(a1, b1), (a2, b2), ..., (an, bn)]

我可以通过索引索引元组中的第一个值。我如何在 xarray 中执行此操作？

Answer 1

Xarray 术语是“multidimensional coordinates”。示例：

import xarray as xr
import numpy as np

data = np.random.random(4)
z = np.arange(4)
x = np.arange(2, 9, 2)
y = np.arange(4) + 100
coords = dict(z=z, x=('z', x), y=('z', y))

arr = xr.DataArray(data, dims=["z"], coords=coords)

而arr是

<xarray.DataArray (z: 4)>
array([0.665017, 0.136371, 0.547075, 0.586285])
Coordinates:
  * z        (z) int64 0 1 2 3
    x        (z) int64 2 4 6 8
    y        (z) int64 100 101 102 103

然后你可以做一些不错的事情，比如

>>> arr.where(arr.y > 100)
<xarray.DataArray (z: 4)>
array([     nan, 0.136371, 0.547075, 0.586285])
Coordinates:
  * z        (z) int64 0 1 2 3
    x        (z) int64 2 4 6 8
    y        (z) int64 100 101 102 103

但我们仍在处理一维对象：

arr.shape  # (4,)
arr.dims   # ('z',)

具有二级索引的一维 xarray

1D xarray with secondary indices

python-xarray