NumPy提出了一种获取数组最大值索引的方法np.argmax.
np.argmax
我想要一个类似的东西,但返回N最大值的索引.
N
例如,如果我有一个数组,[1, 3, 2, 4, 5],function(array, n=3)将返回的索引[4, 3, 1]相对应的元素[5, 4, 3].
[1, 3, 2, 4, 5]
function(array, n=3)
[4, 3, 1]
[5, 4, 3]
python numpy max numpy-ndarray
max ×1
numpy ×1
numpy-ndarray ×1
python ×1