如何找到列向量中n个最大元素的索引,Python
是否有内置函数或非常简单的方法来查找列向量中n个最大元素的索引如何找到列向量中n个最大元素的索引,Python,python,numpy,vector,Python,Numpy,Vector,是否有内置函数或非常简单的方法来查找列向量中n个最大元素的索引 v = [[5] [3] [1] [2] [4]] 找到最大的3个元素的索引 我不止一次地计算重复项,并且输出应该是那些最大数字的索引列表您可以展平数组并获得前n个数字的索引 >>> arr.flatten().argsort()[-3:][::-1] array([0, 4, 1]) 我会使用ravel()而不是flatte()。它的内存效率更高,只需更改视图,而不是创
v = [[5]
[3]
[1]
[2]
[4]]
找到最大的3个元素的索引
我不止一次地计算重复项,并且输出应该是那些最大数字的索引列表您可以展平数组并获得前n个数字的索引
>>> arr.flatten().argsort()[-3:][::-1]
array([0, 4, 1])
我会使用ravel()
而不是flatte()
。它的内存效率更高,只需更改视图,而不是创建另一个数组。