Python递归中保存终端输出的技巧

本文旨在解决在 Python 递归循环中实时显示程序运行状态，并有效保存终端输出的问题。通过介绍 `tqdm` 库的使用方法，结合二分查找的实际案例，详细阐述如何在复杂循环结构中实现进度条的展示，从而提升代码的可观测性和用户体验。

在编写复杂的 Python 程序，尤其是涉及递归或嵌套循环时，了解程序的运行进度至关重要。简单的 print 语句虽然可以输出信息，但频繁的输出可能会刷屏，难以追踪。本文将介绍如何使用 tqdm 库在终端中显示进度条，并结合实际案例，提供一种优雅地展示程序运行状态的方法。

使用 tqdm 显示进度条

tqdm 是一个快速、可扩展的 Python 进度条库，它可以在循环执行过程中动态显示进度。使用 tqdm 非常简单，只需在循环外部初始化 tqdm 对象，并在每次循环迭代后更新进度即可。

安装 tqdm:

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 的参数，我们可以轻松地在各种复杂的循环结构中显示进度条。