从多维列表中选择对应维度中前N大的数值，并返回对应的index

import heapq

a = [[88,1],[75,2], [65,3], [43,4], [52,5]

re1 = heapq.nlargest(4, a, key=lambda x: x[1])  # 求最大的4个元素，并排序
re2 = map(a.index, heapq.nlargest(4, a, key=lambda x: x[1])) 



print(a)
[[88, 1], [75, 2], [65, 3], [43, 4], [52, 5], [98, 6], [77, 7]]

print(re1)
[[77, 7], [98, 6], [52, 5], [43, 4]]

print(list(re2))
[6, 5, 4, 3]