【随手记】python的heapq库的基本用法

简介: 【随手记】python的heapq库的基本用法

每次用都要问AI,干脆记录一下,多用就熟了。


Python的heapq库是用于实现堆(优先队列)算法的库。它提供了一些函数来操作堆结构,如push、pop、heapify等。


下面是一些heapq库常用函数的说明:


  1. heapq.heappush(heap, item):将元素item推入堆heap中。
  2. heapq.heappop(heap):从堆heap中弹出并返回最小的元素。
  3. heapq.heapify(heap):将列表heap原地转换为一个堆。
  4. heapq.heappushpop(heap, item):将元素item推入堆heap,并弹出并返回堆中最小的元素。
  5. heapq.heapreplace(heap, item):弹出并返回堆heap中最小的元素,然后将元素item推入堆中。
  6. heapq.nsmallest(n, iterable):返回可迭代对象iterable中最小的n个元素。
  7. heapq.nlargest(n, iterable):返回可迭代对象iterable中最大的n个元素。


下面是一个示例代码,演示了如何使用heapq来操作堆:


import heapq
# 创建一个空堆
heap = []
# 向堆中添加元素
heapq.heappush(heap, 5)
heapq.heappush(heap, 2)
heapq.heappush(heap, 7)
heapq.heappush(heap, 3)
# 从堆中弹出并返回最小的元素
smallest = heapq.heappop(heap)
print(smallest)  # 输出: 2
# 将列表转换为堆
nums = [6, 1, 9, 4, 8]
heapq.heapify(nums)
print(nums)  # 输出: [1, 4, 6, 9, 8]
# 弹出并返回堆中最小的元素,然后将元素9推入堆中
smallest = heapq.heapreplace(nums, 9)
print(smallest)  # 输出: 1
print(nums)  # 输出: [4, 8, 6, 9, 9]
# 返回列表中最大的3个元素
largest = heapq.nlargest(3, nums)
print(largest)  # 输出: [9, 9, 8]


目录
相关文章
|
3天前
|
机器学习/深度学习 编解码 算法
常用的Python库介绍
Python作为一种功能强大的编程语言,拥有众多的第三方库和框架,这些库和框架覆盖了从数据处理、网络编程、Web开发到人工智能等多个领域。
31 15
|
4天前
|
分布式计算 大数据 Java
如何使用Python的pyodps库来进行跨项目空间重命名表名?
MaxCompute作为一款全面的大数据处理平台,广泛应用于各类大数据分析、数据挖掘、BI及机器学习场景。掌握其核心功能、熟练操作流程、遵循最佳实践,可以帮助用户高效、安全地管理和利用海量数据。以下是一个关于MaxCompute产品使用的合集,涵盖了其核心功能、应用场景、操作流程以及最佳实践等内容。
30 12
|
3天前
|
存储 监控 数据可视化
【Bokeh 库】Python 中的动态数据可视化
【7月更文挑战第15天】Python的Bokeh库是用于动态数据可视化的利器,它能创建交互式、现代Web浏览器兼容的图表。安装Bokeh只需`pip install bokeh`。基础概念包括Plot、Glyph、数据源和工具。通过示例展示了如何用Bokeh创建动态折线图,包括添加HoverTool。Bokeh还支持散点图、柱状图,可自定义样式和布局,添加更多交互工具,并能构建交互式应用和实时数据流更新。适用于数据探索和实时监控。
21 5
|
9天前
|
SQL 并行计算 API
Dask是一个用于并行计算的Python库,它提供了类似于Pandas和NumPy的API,但能够在大型数据集上进行并行计算。
Dask是一个用于并行计算的Python库,它提供了类似于Pandas和NumPy的API,但能够在大型数据集上进行并行计算。
|
9天前
|
机器学习/深度学习 人工智能 数据挖掘
Numba是一个Python库,用于对Python代码进行即时(JIT)编译,以便在硬件上高效执行。
Numba是一个Python库,用于对Python代码进行即时(JIT)编译,以便在硬件上高效执行。
|
9天前
|
网络协议 安全 Shell
`nmap`是一个开源的网络扫描工具,用于发现网络上的设备和服务。Python的`python-nmap`库允许我们在Python脚本中直接使用`nmap`的功能。
`nmap`是一个开源的网络扫描工具,用于发现网络上的设备和服务。Python的`python-nmap`库允许我们在Python脚本中直接使用`nmap`的功能。
|
9天前
|
机器人 Shell 开发者
`roslibpy`是一个Python库,它允许非ROS(Robot Operating System)环境(如Web浏览器、移动应用等)与ROS环境进行交互。通过使用`roslibpy`,开发者可以编写Python代码来远程控制ROS节点,发布和订阅话题,以及调用服务。
`roslibpy`是一个Python库,它允许非ROS(Robot Operating System)环境(如Web浏览器、移动应用等)与ROS环境进行交互。通过使用`roslibpy`,开发者可以编写Python代码来远程控制ROS节点,发布和订阅话题,以及调用服务。
|
6天前
|
数据采集 搜索推荐 机器人
Python 神器:wxauto 库
Python 神器:wxauto 库
41 1
|
9天前
|
自然语言处理 程序员 编译器
`pylatex`是一个Python库,用于生成LaTeX文档。LaTeX是一种用于高质量排版和打印的文档准备系统,特别适用于科学、技术和数学文档。
`pylatex`是一个Python库,用于生成LaTeX文档。LaTeX是一种用于高质量排版和打印的文档准备系统,特别适用于科学、技术和数学文档。
|
9天前
|
存储 对象存储 Python
`openpyxl`是一个用于读写Excel 2010 xlsx/xlsm/xltx/xltm文件的Python库。它不需要Microsoft Excel,也不需要.NET或COM组件。
`openpyxl`是一个用于读写Excel 2010 xlsx/xlsm/xltx/xltm文件的Python库。它不需要Microsoft Excel,也不需要.NET或COM组件。