直接索引arr[index_array]是最简单可靠的高级索引方式,按index_array顺序提取元素,结果形状与索引数组一致;np.take()适用于需显式控轴、处理越界或提升兼容性/可读性场景。
用
直接索引是最简单可靠的方式
NumPy 的高级索引(advanced indexing)允许你用整数数组作为下标一次性取多个元素,结果形状与索引数组一致。这不是“复制后排序”,而是按
中的顺序从原数组中抓取对应位置的值。
常见错误是误以为这等价于
或混淆了
的用途——前者是取值,后者是求排序下标,二者目的不同。
必须是 1D 整型数组,且所有值在
范围内,越界会报
支持重复索引,比如
会返回包含
两次的结果
多维数组也适用:若
是二维的,
沿轴 0 索引;要沿其他轴操作,得用
什么时候该用
而不是直接索引
当需要显式控制轴、处理越界行为,或在构建通用函数时增强可读性,
更合适。它本质是封装了高级索引,但多了几个关键参数:
:明确指定在哪一维上应用索引,避免对高维数组写
这类易错表达
:可设为
(自动截断到合法范围)或
(循环取模),而直接索引遇到越界直接报错
在 JIT 编译(如 Numba)或某些旧版 NumPy 中,
兼容性更稳,直接索引可能触发隐式拷贝或广播警告
示例:
在长度为 4 的数组上等价于取
。
立即学习
“
Python免费学习笔记(深入)
”;
Python 3.14.3
微软官方的 Python 扩展,是 VS Code 安装量最高的扩展(209M+)。集成 IntelliSense(通过 Pylance)、调试(通过 Python Debugger)、代码检查、格式化、重构和单元测试等功能。支持 Jupyter Notebook、虚拟环境管理和多 Python 版本切换。
下载
别把
和索引重组混为一谈
返回的是“排序后各位置对应原数组的下标”,不是重组用的索引数组。它常被误当作重排工具,但实际用途是构造排序映射。
若想按某列排序二维数组并重组行:先用
,再用
—— 这里
才是真正的重组索引
直接写
对一维数组虽能排序,但掩盖了“先算索引、再取值”两步逻辑,不利于调试或条件重组
若索引数组来自外部(比如用户输入、文件读取),千万别先
再用——那是在对索引本身排序,不是按索引取值
高维重组容易忽略的轴对齐问题
对形状为
的二维数组,用 1D 索引数组
时,默认作用于第 0 轴(行)。如果想按列重排(即固定某行,重排列顺序),必须显式指定轴或使用花式索引语法。
错误写法:
→ 重排行,不是列
正确做法一:
→ 沿列轴取
正确做法二:
→ 显式切片,语义清晰,但要求
是 1D
若
是二维(如每行用不同列序),就得用
,这时容易因广播维度搞错形状
这种轴意识一旦松懈,输出形状就和预期差一个转置,且不容易一眼看出来。
arr[index_array]index_arraynp.take()np.argsort()index_array[0, len(arr))IndexError: index N is out of bounds for axis 0 with size Marr[[2, 2, 0]]arr[2]arrarr[idx]np.take(arr, idx, axis=1)np.take()np.take()axisarr[:, idx]mode'clip''wrap'np.take()np.take(arr, [−1, 0, 5], mode='clip')[3, 0, 3]np.argsort()np.argsort()idx = np.argsort(arr[:, col])arr[idx]idxarr[np.argsort(arr)]np.argsort()(N, M)idxarr[idx]np.take(arr, idx, axis=1)arr[:, idx]idxidxarr[np.arange(arr.shape[0])[:, None], idx]