本文旨在解决在 Python 递归循环中实时显示程序运行状态，并有效保存终端输出的问题。通过介绍 `tqdm` 库的使用方法，结合二分查找的实际案例，详细阐述如何在复杂循环结构中实现进度条的展示，从而提升代码的可观测性和用户体验。
在编写复杂的 Python 程序，尤其是涉及递归或嵌套循环时，了解程序的运行进度至关重要。简单的 print 语句虽然可以输出信息，但频繁的输出可能会刷屏，难以追踪。本文将介绍如何使用 tqdm 库在终端中显示进度条，并结合实际案例，提供一种优雅地展示程序运行状态的方法。
使用 tqdm 显示进度条
tqdm 是一个快速、可扩展的 Python 进度条库，它可以在循环执行过程中动态显示进度。使用 tqdm 非常简单，只需在循环外部初始化 tqdm 对象，并在每次循环迭代后更新进度即可。
安装 tqdm:
立即学习“Python免费学习笔记（深入）”；
pip install tqdm
基本用法:
from tqdm import tqdm
import time
for i in tqdm(range(100)):
# 模拟耗时操作
time.sleep(0.01)上述代码将在终端中显示一个进度条，随着循环的进行，进度条会不断更新，直到循环结束。
在复杂循环中使用 tqdm
当循环结构比较复杂，例如嵌套循环或递归循环时，tqdm 同样可以发挥作用。关键在于合理地设置 tqdm 的参数，例如 total 和 desc。
- total：指定循环的总迭代次数，tqdm 根据这个值计算进度百分比。
- desc：指定进度条的描述信息，可以用于区分不同的循环。
- leave：指定循环完成后是否保留进度条。
示例：嵌套循环中的 tqdm
from tqdm import tqdm
import time
for i in tqdm(range(10), desc="Outer Loop", leave=False):
for j in tqdm(range(5), desc=f"Inner Loop (i={i})", leave=False):
# 模拟耗时操作
time.sleep(0.1)在这个例子中，外层循环和内层循环都使用了 tqdm，并分别设置了不同的描述信息。leave=False 表示循环结束后不保留进度条，可以避免终端输出过于冗余。
实际案例：二分查找中的 tqdm
二分查找是一种常见的算法，其迭代次数取决于查找范围和精度要求。我们可以利用 tqdm 在二分查找过程中显示进度。
from tqdm import tqdm
import math
import time
def binary_search(low, high, tolerance, target_function):
"""
使用二分查找求解方程的根。
Args:
low: 查找范围的下界。
high: 查找范围的上界。
tolerance: 精度要求。
target_function: 目标函数，例如 f(x) = x^2 - 2。
Returns:
近似解。
"""
# 计算最大迭代次数
max_iterations = math.ceil(math.log2((high - low) / tolerance))
with tqdm(total=max_iterations, desc="Binary Search", leave=False) as pbar:
while abs(high - low) > tolerance:
mid = (high + low) / 2
if target_function(mid) > 0:
high = mid
else:
low = mid
pbar.update(1)
time.sleep(0.05) # 模拟计算耗时
return (high + low) / 2
# 示例：求解 x^2 - 2 = 0 的根
def target_function(x):
return x**2 - 2
# 设置查找范围和精度
low = 1.0
high = 2.0
tolerance = 1e-5
# 进行二分查找
root = binary_search(low, high, tolerance, target_function)
print(f"Approximate root: {root}")在这个例子中，我们首先计算了二分查找的最大迭代次数，然后使用 tqdm 创建了一个进度条。在每次迭代中，我们更新进度条，直到满足精度要求。
注意事项与总结
- tqdm 适用于迭代次数可预测的循环。如果迭代次数无法提前确定，可以使用 tqdm 的 update() 方法手动更新进度。
- 在多线程或多进程程序中，需要注意 tqdm 的线程安全问题。可以使用 tqdm.lock 来保证线程安全。
- tqdm 的输出会干扰其他终端输出。可以使用 tqdm.write() 方法输出信息，避免干扰。
- 在 Jupyter Notebook 中，tqdm 会自动显示交互式进度条。
总而言之，tqdm 是一个非常实用的 Python 库，可以帮助我们更好地了解程序的运行状态，提升代码的可观测性和用户体验。通过合理地设置 tqdm 的参数，我们可以轻松地在各种复杂的循环结构中显示进度条。
