如何在xarray中从指定网格位置选取周边X个网格点范围内的所有点？

我有一个xarray数据集ds，其维度为['time','lat','lon']，需要指定一个参考网格点ref_grid_point，并选取该点周边X个网格点范围内的所有网格点（X为整数，代表距离参考点的网格点数量，比如1、2…n）。

之前如果以度数作为选取半径，可以用sel配合slice实现：

ds.sel(lat = slice(ref_grid_point['lat'] + radius, ref_grid_point['lat'] - radius), 
       lon = slice(ref_grid_point['lon'] - radius, ref_grid_point['lon'] + radius))

以下是按网格点数量选取的实现步骤：

• 步骤1：获取参考点的维度索引
  先定位参考点在lat和lon维度上的索引位置：

  # 获取参考点的纬度索引
  ref_lat_idx = ds.lat.where(ds.lat == ref_grid_point['lat'], drop=True).index[0]
  # 获取参考点的经度索引
  ref_lon_idx = ds.lon.where(ds.lon == ref_grid_point['lon'], drop=True).index[0]
  

• 步骤2：计算索引范围（处理边界越界）
  根据设定的X值计算选取的索引上下界，同时确保索引不会超出数据集的维度范围：

  X = 2  # 示例：选取周边2个网格点的范围
  # 计算纬度索引的有效范围
  lat_min_idx = max(0, ref_lat_idx - X)
  lat_max_idx = min(len(ds.lat) - 1, ref_lat_idx + X)
  # 计算经度索引的有效范围
  lon_min_idx = max(0, ref_lon_idx - X)
  lon_max_idx = min(len(ds.lon) - 1, ref_lon_idx + X)
  

• 步骤3：按索引选取区域
  使用isel方法基于索引切片获取目标区域：

  selected_ds = ds.isel(
      lat=slice(lat_min_idx, lat_max_idx + 1), 
      lon=slice(lon_min_idx, lon_max_idx + 1)
  )
  

  注：由于Python的slice是左闭右开区间，所以上限需要加1，确保包含ref_lat_idx + X和ref_lon_idx + X对应的网格点。

• 简化写法（无需边界处理时）
  如果能确定X值不会导致索引超出数据集范围，可以直接简化为：

  selected_ds = ds.isel(
      lat=slice(ref_lat_idx - X, ref_lat_idx + X + 1),
      lon=slice(ref_lon_idx - X, ref_lon_idx + X + 1)
  )
  

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