如何有效地迭代 Python 列表中重叠的元素对（或三元组）？

在 Python 中迭代重叠值对

在 Python 中迭代列表时，您可能需要访问当前元素和同时下一个元素。传统上，这是使用以下代码完成的：

for current, next in zip(the_list, the_list[1:]):
    # Do something

但是，Python 3.8 引入了一种更有效的方法来实现此目的：

利用配对函数

Python 文档为此提供了一个简单的成对函数目的：

import itertools

def pairwise(iterable):
    a, b = itertools.tee(iterable)
    next(b, None)
    return zip(a, b)

该函数创建两个迭代器，a和b，它们指向输入可迭代的第一个元素。然后迭代器 b 前进一步，导致 a 指向当前元素，b 指向下一个元素。然后使用 zip 函数创建这些元素对。

对于 Python 2

对于 Python 2，您可以将类似的成对函数与 itertools.izip 一起使用函数而不是 zip：

import itertools

def pairwise(iterable):
    a, b = itertools.tee(iterable)
    next(b, None)
    return itertools.izip(a, b)

泛化为多个Elements

通过调整 tee 调用中的 n 参数，可以将成对函数推广为生成更大的元素窗口。例如，要创建三个元素对，您可以使用：

def threes(iterator):
    a, b, c = itertools.tee(iterator, 3)
    next(b, None)
    next(c, None)
    next(c, None)
    return zip(a, b, c)

Caveat

请务必注意，此技术可能会消耗大量内存如果一个迭代器比其他迭代器前进得更远。当窗口尺寸很大或者原始可迭代中有很多元素时，就会发生这种情况。

以上是如何有效地迭代 Python 列表中重叠的元素对（或三元组）？的详细内容。更多信息请关注PHP中文网其他相关文章！