如何在xarray中实现类似Pandas的布尔索引功能？

针对你的需求，xarray提供了两种常用方法来实现类似Pandas的布尔索引筛选，具体操作如下：

1. 先构造示例Dataset #

首先创建对应你示例的xarray Dataset：

import xarray as xr

data = {
    'A': ('index', [5, 4, 3, 4, 5]),
    'B': ('index', [1, 1, 5, 3, 2]),
    'C': ('index', [0, 0, 0, 0, 0])
}
ds = xr.Dataset(data)

2. 方法一：使用where()方法 #

where()会保留符合条件的元素，不符合的默认设为NaN，添加drop=True参数可直接删除不符合条件的维度条目：

# 筛选B列值大于2的行，删除不符合的条目
ds_filtered = ds.where(ds['B'] > 2, drop=True)

3. 方法二：使用isel()结合布尔掩码 #

先创建布尔掩码，再通过isel()选择对应维度的索引位置：

# 创建布尔掩码
mask = ds['B'] > 2
# 基于掩码筛选index维度的条目
ds_filtered = ds.isel(index=mask)

筛选结果 #

执行上述任意一种方法后，得到的Dataset为：

<xarray.Dataset>
Dimensions:  (index: 2)
Coordinates:
  * index    (index) int64 2 3
Data variables:
    A        (index) int64 3 4
    B        (index) int64 5 3
    C        (index) int64 0 0

内容的提问来源于stack exchange，提问作者Billiam