写代码从 NumPy 数组中获取前 N 个最大值的索引。

Write a code to get the indices of N maximum values from a NumPy array.

题目类型: 技术面试题

这是一道技术面试题，常见于澳洲IT公司面试中。

难度: medium

分类: numpy

标签: numpy, argsort, top-n

参考答案摘要

TL;DR 使用 np.argsort 排序索引并切片取最后 N 个。 import numpy as np arr = np.array([1, 5, 2, 9, 7]) N = 3 idx = np.argsort(arr)[-N:][::-1] print(idx)

本题提供 STAR 原则详细解答和技术解析，登录匠人学院学习中心即可查看完整答案。