2D XArray cfgrib引擎索引

Question

我想在lat/long窗口内从grib2文件中提取数据。

def get_data(temp_filename):
    ds = xr.open_dataset(temp_filename, engine="cfgrib")
    ds = ds[{'latitude': ds.latitude > 1, 'longitude': ds.longitude > -100}]
    radar_values = ds.to_array()
    print(radar_values)

给出

IndexError: too many indices for array

Answer 1

事实证明，该错误与使用xarray执行2D索引有关。

d = ds[{'latitude': ds.latitude > 1, 'longitude': ds.longitude > -100}]

应该使用where()方法来完成

ds=ds.where(ds.latitude > 1)
ds=ds.where(ds.longitude > -100)

或者通过使用numpy提取底层数组并进行适当的索引。

请参阅http://xarray.pydata.org/en/stable/indexing.html