常用模块和方法记录--heapq模块

heapq模块:heapq 模块提供了堆算法。heapq是一种子节点和父节点排序的树形数据结构

heapq.heapify(ls):heapq模块实现了一个适用于Python列表的最小根堆排序算法。

heapq.heappop(ls):按从右往左pop出去(即去掉的是根节点的数)

heapq.heappush(ls, item):既是小根堆了,那么插入数据,子节点就必定比父节点要大的

heapq.heapreplace(ls, val):ls为列表,val为插入值(pop出ls[0]并取代他的位置)

heapq.pushpop(heap,item):pop出根节点,然后再push item进去,仍旧按照小根堆的排序方式

heapq.merge(*itemables):合并多个堆然后输出,返回的是一个对象,需要用list方法变成列表才能看见结果

举例设有ls  =[1,2,3,4,5,5,6,7,8,8]

heapq.nlargest(n, ls):返回列表中的n个最大值

即heapq.nlargest(4, ls)

返回的是一个列表[8,8,7,6]

heapq.nsmallest(n, ls):返回列表中的n个最小值

即heapq.nsmallest(4, ls)

返回的是一个列表[1,2,3,4]

这个模块在涉及到求前几个最大or最小的数用的比较多,也是一个比较常用的模块

猜你喜欢

转载自blog.csdn.net/honorwh/article/details/89022686
今日推荐