NumPy提出了一种获取数组最大值索引的方法np.argmax.
我想要一个类似的东西,但返回N最大值的索引.
例如,如果我有一个数组,[1, 3, 2, 4, 5],function(array, n=3)将返回的索引[4, 3, 1]相对应的元素[5, 4, 3].
假设你有一个numpy向量[0,3,1,1,1],你运行argsort
你会得到[0,2,3,4,1]但所有的都是一样的!我想要的是一种有效的方法来改变相同值的索引.有没有想法如何在没有带有两个索引的while循环的情况下做到这一点?
numpy.array([0,3,1,1,1]).argsort()
Run Code Online (Sandbox Code Playgroud) 使用python在大小为N的未排序列表中获取k个最小数字的最快方法是什么?
对大数字列表进行排序,然后获得k个最小数字,
或者通过在列表中找到k次中的最小值来获得k个最小数字,确保在下一个搜索之前从搜索中删除找到的最小值?