推荐用 np.argpartition 获取前k大值索引，时间复杂度 o(n)，仅保证第k位左侧为大于等于元素；返回索引未按值排序，需二次 argsort 实现降序；二维需指定 axis 并配合 take_along_axis。
用 np.argpartition 快速获取前K大值索引（推荐）
直接调用 np.argpartition 是最高效的方式，它不完全排序，只保证第K个位置左边是“大于等于”的元素，时间复杂度接近 O(n)。注意：返回的索引对应的是原数组中前K大值的位置，但这些索引本身不按值大小排序。
常见错误是误以为 np.argpartition(arr, -k) 返回的前k个索引已按对应值降序排列——其实不是，它们只是“属于前K大”的一组索引，顺序是随机的。
- 对一维数组
arr，取前K大值索引：indices = np.argpartition(arr, -k)[-k:] - 若需按对应值从大到小排序索引，再补一步：
indices = indices[np.argsort(arr[indices])[::-1]] - K 不能超过数组长度，否则会报
IndexError；建议先做min(k, len(arr))安全截断
用 np.argsort 全排序（简单但慢）
当 K 接近数组长度，或你顺带需要所有元素的完整排名时，np.argsort 更直观。它返回升序索引，所以前K大值索引就是末尾K个反转过来。
性能上，O(n log n) 明显比 argpartition 差，尤其在大数组、小K场景下浪费明显。
SkillSub Pro - Python 题解与代码注释双功能技能功能概述SkillSub Pro - Python 题解与代码注释双功能技能是一项面向实际任务的技能，主要用于SkillSub Pro 是一个 Python 题解生成与代码注释的 双功能合体技能 ，专为学生、算法学习者和开发者设计；✅ 一个技能，两种用途 ：；核心要点📝 题解模式 ：输入题目/题号，自动生成完整 Python 题解（含详细注释、解题思路、复杂度分析）；💬 注释模式 ：输入 Python 代码，自动添加详细中。它将相关步骤、
- 一维情况：
indices = np.argsort(arr)[-k:][::-1] - 二维数组按行取每行前K大索引？不能直接用
argsort而不指定axis；得写循环或用np.take_along_axis配合argsort - 注意
argsort默认 axis=-1，多维时务必确认维度意图
处理二维数组：按行/列找前K大索引
NumPy 没有内置的“每行取 top-K 索引”函数，必须组合使用 argpartition 或 argsort 加 np.arange 索引广播。
最常用的是按行操作（如：每个样本取响应最强的K个特征索引）。关键点在于避免 Python 循环，用向量化方式构造高级索引。
- 按行取每行前K大索引（返回 shape=(n_rows, k) 的索引数组）：
indices_2d = np.argpartition(arr, -k, axis=1)[:, -k:]<br>sorted_indices = np.take_along_axis(indices_2d, np.argsort(np.take_along_axis(arr, indices_2d, axis=1))[:, ::-1], axis=1)
- 更稳妥的做法是先用
argpartition粗筛，再对每个子集局部argsort——虽然略啰嗦，但可读性和稳定性更好 - 如果数组含
NaN，argpartition和argsort都会把NaN排在最后（取决于nan_policy），但 NumPy 默认不提供该参数；需提前用np.nan_to_num或掩码过滤
边界与兼容性注意点
不同 NumPy 版本对负索引和 argpartition 边界行为基本一致，但 K=0 或空数组仍可能触发意外行为。
- K=0 时，
argpartition(arr, 0)合法，但[-0:]得空数组，需显式判断if k == 0: return np.array([]) - 输入是
float32时，argpartition结果与float64一致，但排序稳定性略有差异（不影响索引正确性） - 若后续要把这些索引用于高级索引（如
arr[indices]），确保indices是整数类型（np.intp最安全），避免FutureWarning: Using a non-tuple sequence for multidimensional indexing
真正容易被忽略的是：当你用 argpartition 得到索引后直接传给 plt.scatter 或其他绘图函数时，坐标轴可能错位——因为索引没重排，画出来的点不按值大小顺序排列，视觉上显得“乱”。这时候不加 argsort 二次排序，就达不到“标出最大K个”的预期效果。
Python免费学习笔记（深入）：立即使用
在学习笔记中，你将探索 Python 的核心概念和高级技巧！
