在xarray datafram中过滤2坐标索引

2024-09-28 23:16:20 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试筛选xarray中的大型数据集,以从以下数据集中获得精确的latitudelongitude值:

<xarray.Dataset>
Dimensions:    (latitude: 23, level: 6, longitude: 21, time: 178486)
Coordinates:
  * time       (time) datetime64[ns] 1979-01-01 ... 2019-11-26T21:00:00
  * latitude   (latitude) float32 46.5 46.25 46.0 45.75 ... 41.5 41.25 41.0
  * longitude  (longitude) float32 18.0 18.25 18.5 18.75 ... 22.5 22.75 23.0
  * level      (level) int32 750 800 850 900 950 1000
Data variables:
    cbh        (time, latitude, longitude) float32 ...
    clwc       (time, level, latitude, longitude) float32 ...
    t          (time, level, latitude, longitude) float32 ...
    vetar      (time, level, latitude, longitude) float32 ...
    sp         (time, latitude, longitude) float32 ...
Attributes:
    Conventions:  CF-1.6
    history:      2019-05-11 06:14:51 GMT by grib_to_netcdf-2.10.0: /opt/ecmw...

我正试图用where语句来实现,但似乎我需要比较数组来实现这一点。 使用DS1.where(DS1.longitude==22.0 and DS1.latitude==43.5,drop=True)我得到了一个著名的错误:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我可以通过两个步骤执行此过滤,第一步是

ds22=DS1.where(DS1.longitude==22.0,drop=True) 

然后是

ds22435=ds22.where(ds22.latitude==43.5,drop=True)

但有没有办法一步到位呢


Tags: 数据truetimewhereleveldatasetdropdimensions