如何设置numpy数组相对于起始索引的结束索引

Numpy 原生的切片语法不直接支持「起始索引+采样数量」的简写，你示例中的a[3:+4]写法无法得到预期结果：切片中单独写的+4会被解析为相对于数组起始位置的索引4，实际等价于a[3:4]，仅能取到索引为3的1个元素。

如果要实现从指定位置开始提取固定数量元素的需求，可以用以下几种方式实现：

• 方式1：直接计算结束索引，这是最常用、性能无损耗的写法
  代码示例：

  import numpy as np
  a = np.arange(10)
  # 从索引3开始，提取4个元素，结束索引=3+4=7
  c = a[3:3+4] 
  # 输出结果为 [3 4 5 6]，和a[3:7]效果完全一致
  

• 方式2：封装为通用工具函数，适合频繁使用该逻辑的场景
  代码示例：

  def slice_by_count(arr: np.ndarray, start_idx: int, count: int):
      # 可自行添加边界判断逻辑，避免索引越界
      end_idx = min(start_idx + count, len(arr))
      return arr[start_idx:end_idx]
  
  # 调用示例
  c = slice_by_count(a, 3, 4)
  

• 高维数组场景也可以沿用相同逻辑，比如二维数组从(x,y)位置开始取n行m列：

  # 示例：提取从(1,2)开始的3行2列数据
  arr_2d = np.arange(5*5).reshape(5,5)
  res = arr_2d[1:1+3, 2:2+2]
  

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