获取列表中的最大的N项和最小的N项

获取列表中的最大的N项和最小的N项

#!/sur/bin/env python
# -*- coding:utf-8 -*-
# author:zengsf
#time:2018/10/31

import heapq

nums = [1, 0, 2, 29, 7, -7, 18, 23, 5, 20, 9, 10, 12]

# 最大的4个数的索引
max_num_index_list = map(nums.index, heapq.nlargest(4, nums))

# 最小的4个数的索引
min_num_index_list = map(nums.index, heapq.nsmallest(4, nums))

print(max_num_index_list)
print(list(max_num_index_list))
print(list(min_num_index_list))

输出结果：

<map object at 0x0000021066E627F0>
[3, 7, 9, 6]
[5, 1, 0, 2]