得到列表前n个值的方法有很多,本文只介绍两种相对简单的方法。
方法一:
使用heapq库:
#如果列表的值各不相同,则可正确获得前n个值的索引
import heapq
list1=[3,2,4,1]
list(map((list1).index, heapq.nlargest(2,list1)))
注意:这种会导致相同的值的索引都以值第一次出现的索引为准。
方法二:
使用numpy库
import numpy as np
list2=[4,5,6,3,2,4,1]
##从小到大排的索引
list(np.argsort(np.array(list2)))