如何利用xarray数据集实现numpy索引

Question

我知道2D数组的x和y索引（numpy索引）

在这个documentation之后，xarray使用例如Fortran风格的索引

所以当我通过的时候

ind_x = [1, 2]
ind_y = [3, 4]

我期望索引对（1,3）和（2,4）有2个值，但xarray返回一个2x2矩阵

现在我想知道如何使用xarray实现类似numpy的索引

注意：我希望避免将整个数据加载到内存中。因此，使用.valuesapi不是我正在寻找的解决方案的一部分

Answer 1

为了考虑速度，我用不同的方法进行了测试

def method_1(file_paths: List[Path], indices) -> List[np.array]:
    data=[]
    for file in file_paths:
        d = Dataset(file, 'r')
        data.append(d.variables['hrv'][indices])
        d.close()
    return data


def method_2(file_paths: List[Path], indices) -> List[np.array]:
    data=[]
    for file in file_paths:
        data.append(xarray.open_dataset(file, engine='h5netcdf').hrv.values[indices])
    return data


def method_3(file_paths: List[Path], indices) -> List[np.array]:
    data=[]
    for file in file_paths:
        data.append(xarray.open_mfdataset([file], engine='h5netcdf').hrv.data.vindex[indices].compute())
    return data

In [1]: len(file_paths)
Out[1]: 4813

结果是：

• 方法_1（使用netcdf4库）：101.9s
• 方法2（使用xarray和API值）：591.4s
• 方法_3（使用xarray+dask）：688.7s

我猜xarray+dask在.compute步中花费了很多时间