Python中bisect模块与堆操作详解

 更新时间:2023年06月07日 09:04:38   作者:郝学胜  
在Python中,bisect和heapq都是处理有序序列的常见模块,这篇文章将分别介绍这两个模块的用法和实现方式,感兴趣的小伙伴可以跟随小编一起学习一下

在Python中,bisectheapq都是处理有序序列的常见模块,本文将分别介绍这两个模块的用法和实现方式。

bisect模块

bisect模块提供了一些函数,实现了对一个序列进行排序及维护已排序序列的一个二分查找。具体来说,bisect模块主要提供了两个函数:bisectinsort

bisect函数

bisect函数是一个通用的二分查找工具,它可以用来查找一个元素在已排序的序列中应该插入的位置,以维持序列的排序。具体用法如下:

import bisect

# 初始化一个已排序的列表
lst = [1, 3, 4, 5, 7, 9]

# 使用bisect函数查找元素插入位置
pos = bisect.bisect(lst, 6)
print(pos)

运行这段代码可以发现,变量pos输出的是4,即元素6应该插入到列表中索引为4的位置,这个列表就能保持升序排列。

insort函数

insort函数是bisect函数的变种,它可以在查找到插入位置的同时插入元素。具体用法和bisect类似,只是调用的函数不同,如下:

import bisect

# 初始化一个已排序的列表

lst = \[1, 3, 4, 5, 7, 9]

# 使用insort函数插入元素

bisect.insort(lst, 6)
print(lst)

运行这段代码可以得到输出:

\[1, 3, 4, 5, 6, 7, 9]

这个函数接受两个参数,第一个参数是已排序的序列,第二个参数是要插入的元素。该函数会将元素插入到序列的正确位置。

heapq模块

heapq模块是一个堆队列算法模块,提供了在列表上进行堆操作的函数。下面我们来看一下heapq模块如何实现堆。

初始化堆

首先,初始化一个空堆和向堆中插入元素的方法如下:

import heapq

# 初始化一个空堆

heap = \[]

# 往堆中插入元素

heapq.heappush(heap, 3)
heapq.heappush(heap, 1)
heapq.heappush(heap, 7)
heapq.heappush(heap, 6)

以上代码先用heap的索引为0的位置存储根节点的值,以此类推往下存储每个节点的值,然后在heap中插入元素使用的是heapq.heappush函数。插入后的堆如下图所示:

  1
   \
   3
    \
    7
   /
  6

弹出元素

从堆中弹出元素使用的是heapq.heappop函数,由于我们使用的是最小堆,根节点的值是最小的。弹出元素的代码如下:

# 从堆中弹出元素

print(heapq.heappop(heap))
print(heapq.heappop(heap))
print(heapq.heappop(heap))
print(heapq.heappop(heap))

运行以上代码可以看到输出:

1
3
6
7

这里的堆是使用数组实现的最小堆(根节点的值是最小的),如果要实现最大堆,可以在插入元素时取反,在弹出元素时再取反。

总结

Python的bisect模块和heapq模块分别用来操作有序序列和堆。bisect模块提供了二分查找和插入元素的函数,heapq模块则提供了堆操作的函数。熟练使用这些模块能够提高编码效率,并加深对数据结构和算法的理解。

到此这篇关于Python中bisect模块与堆操作详解的文章就介绍到这了,更多相关Python bisect heapq内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python中往列表中插入字典时,数据重复问题

    Python中往列表中插入字典时,数据重复问题

    这篇文章主要介绍了Python中往列表中插入字典时,数据重复问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2024-02-02
  • 打包python 加icon 去掉cmd黑窗口方法

    打包python 加icon 去掉cmd黑窗口方法

    今天小编就为大家分享一篇打包python 加icon 去掉cmd黑窗口方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • PyCharm调用matplotlib绘图时图像弹出问题详解

    PyCharm调用matplotlib绘图时图像弹出问题详解

    这篇文章主要给大家介绍了关于PyCharm调用matplotlib绘图时图像弹出问题的相关资料,文中通过图文介绍的非常详细,对大家学习或者使用PyCharm具有一定的参考学习价值,需要的朋友可以参考下
    2022-07-07
  • 使用Python批量生成PPT版荣誉证书的示例代码

    使用Python批量生成PPT版荣誉证书的示例代码

    使用Python处理PPT文件通常需要使用第三方库来简化对PPT文件的读取、写入和修改操作,本文将给大家介绍一个小案例,使用Python批量生成PPT版荣誉证书,感兴趣的同学跟着小编一起来看看吧
    2023-08-08
  • Django零基础入门之路由path和re_path详解

    Django零基础入门之路由path和re_path详解

    这篇文章主要介绍了Django零基础入门之路由path和re_path,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-09-09
  • Python使用pyglet库完整实现汉诺塔游戏流程详解

    Python使用pyglet库完整实现汉诺塔游戏流程详解

    这篇文章主要介绍了Python使用pyglet库完整实现汉诺塔游戏流程,汉诺塔问题是一个递归问题,也可以使用非递归法来解决,这个问题不仅是一个数学和逻辑问题,也是一个很好的教学工具,可以用来教授递归、算法和逻辑思考等概念,需要的朋友可以参考下
    2007-02-02
  • Python sklearn中的K-Means聚类使用方法浅析

    Python sklearn中的K-Means聚类使用方法浅析

    这篇文章主要介绍了Python sklearn中的K-Means聚类使用方法,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2022-12-12
  • pytorch 中forward 的用法与解释说明

    pytorch 中forward 的用法与解释说明

    这篇文章主要介绍了pytorch 中forward 的用法与解释说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-02-02
  • Python基础教程之if判断,while循环,循环嵌套

    Python基础教程之if判断,while循环,循环嵌套

    这篇文章主要介绍了Python基础教程之if判断,while循环,循环嵌套 的相关知识,非常不错,具有一定的参考借鉴价值,需要的朋友可以参考下
    2019-04-04
  • Python爬取微信读书实现读书免费自由

    Python爬取微信读书实现读书免费自由

    主要跟大家介绍一下,我是如何用Python爬取小说,再导入微信读书的。成功实现在微信读书中各种“白票”付费小说,有需要的朋友可以借鉴参考下
    2021-09-09

最新评论