【python】堆heap

记录序列的前5大

import heapq
list_o = [1,6,4,9,10,8,4]
list_s = []
for num in list_o:        
    heapq.heappush(list_s, num)
    if(len(list_s) > 5):
        heapq.heappop(list_s)
print(list_s)