使用slice对象列表切片NumPy数组时触发IndexError的原因与解决

错误原因 #

NumPy对索引的解析规则里，列表和元组的处理逻辑完全不同：

• 用元组作为索引时，元组内每个元素对应数组的一个维度，用来指定该维度的切片/索引规则，这是多维切片的标准用法。
• 用列表作为索引时，NumPy会触发花式索引（Fancy Indexing），此时列表里的元素会被当作对第一个维度的离散索引（比如a[[0,1]]是取第0、1行）。而slice对象不能作为花式索引的元素，因此直接抛出IndexError。

你看到的medpy代码能正常运行，是因为早期NumPy版本中，赋值操作的左侧索引对列表兼容性更好，会自动将列表转为元组处理；但在取值操作（比如你写的b = a[slicer]）中，列表会严格触发花式索引规则，导致报错。

修复方法 #

把切片对象的列表转为元组即可，有两种实现方式：

方式1：直接定义为元组 #

import numpy as np
a = np.array([[1,2,3], [4,5,6], [7,8,9]])
slicer = (slice(0,2), slice(0,2))  # 用圆括号定义元组
b = a[slicer]

方式2：将现有列表转为元组 #

如果已经有列表形式的slicer，用tuple()转换：

slicer = [slice(0,2), slice(0,2)]
b = a[tuple(slicer)]

运行后b会得到预期的2x2子数组：

[4, 5]])```

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