80、Python 迭代工具与性能测量：从基础到实践

Python 迭代工具与性能测量：从基础到实践

1. 单次迭代与 myzip 函数

在 Python 中，有些内置函数（如 map ）仅支持单次迭代，迭代结束后结果就为空。以模拟 zip 函数的 myzip 为例：

def myzip(*args):
    iters = map(iter, args)
    while iters:
        res = [next(i) for i in iters]
        yield tuple(res)

此代码利用 iter 和 next 函数，可与任何可迭代对象协作。无需捕获 next(i) 引发的 StopIteration 异常，因为它会使生成器停止工作，就像使用 return 语句一样。 while iters: 语句足以遍历参数元素，前提是至少传入一个参数，同时能避免无参数时的无限循环。

该代码在 Python 2.x 中运行正常：

>>> list(myzip('abc', 'lmnop'))
[('a', 'l'), ('b', 'm'), ('c', 'n')]

但在 Pyt