写一段代码,从 NumPy 数组中获取 N 个最大值的索引。
Write a code to get the indices of N maximum values from a NumPy array.
题目类型: 技术面试题
这是一道技术面试题,常见于澳洲IT公司面试中。
难度: medium
分类: Python
标签: numpy, argsort, index
参考答案摘要
答案 我们可以使用下面的代码从 NumPy 数组中获取 N 个最大值的索引: import numpy as np ar = np.array([1, 3, 2, 4, 5, 6]) print(ar.argsort()[-3:][::-1])
本题提供 STAR 原则详细解答和技术解析,登录匠人学院学习中心即可查看完整答案。