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