基於坐標的xarray選擇

Question

我有一堆數據要插入到xarray數據xarray 。 數據集下方只有一個 DataArray，但將來可能會更多。 一些數據集維度被注釋，我已經將它們作為沿着一個或多個軸的坐標插入到數據集中。 現在我想獲得一個坐標annot1 = '1'的數據集。

我用ds.sel()方法嘗試了一堆不同的字典，但這似乎不是解決方案。 我將如何以這樣一種方式索引以下數據集，以便我只獲取annot1 = '1' ？

import xarray as xr
import numpy as 

coords = {
    'x':['a','b','c','d'], 
    'y':['A','B','C','D'], 
    'z':['d_000','d_001','d_010','d_011','d_100','d_101'],
    'annot1': ('z',['0','1','0','1','0','1']),
    'annot2': ('z',['0','0','1','1','0','0']),
    'annot3': ('z',['0','0','0','0','1','1']),
    'intens':(['x','y'], np.arange(4**2).reshape((4,4))),
}


ds = xr.Dataset(
    data_vars={
        'data': (['x','y','z'], np.random.randn(4,4,6))
    },
    coords=coords,
)
ds.sel(annot1 = '1') # How do i select only the data where annot1 is '1'?

Answer 1

找到了我要找的東西： ds.sel(z=ds['annot1']=='1')