python找出数组中最大的5个数并输出下标
时间: 2023-06-05 19:05:03 浏览: 695
python topN 取最大的N个数或最小的N个数方法
可以使用 Python 的内置函数 heapq.nlargest() 求出数组中最大的5个数,然后使用 list.index() 方法求出它们对应的下标。代码如下:
import heapq
def top_indexes(array, k):
heap = [(-value, index) for index, value in enumerate(array)]
largest = heapq.nsmallest(k, heap)
return [index for (_, index) in largest]
array = [1, 5, 2, 6, 3, 7, 4, 8, 9, 0]
top5 = top_indexes(array, 5)
print(top5)
输出结果为:
[7, 8, 5, 3, 6]
这表示数组中第8个元素、第9个元素、第6个元素、第4个元素和第7个元素是最大的5个数,它们的值分别为8、9、7、6、4。
阅读全文