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

在 Python 中迭代重叠值对

在 Python 中迭代列表，同时考虑“当前”和“下一个”元素是一个共同的需要。传统上，这是使用以下代码实现的：

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

虽然有效，但这种方法并不是最惯用或最有效的。更Pythonic的解决方案是使用itertools模块中的pairwise函数：

import itertools

def pairwise(iterable):
    "s -> (s0, s1), (s1, s2), (s2, s3), ..."
    a, b = itertools.tee(iterable)
    next(b, None)
    return zip(a, b)

pairwise函数创建两个并行迭代器，a和b，都指向原始可迭代的第一个元素。然后将 b 前进一步，导致 a 指向 s0，b 指向 s1。然后 zip 函数将 a 和 b 中的元素配对，为您提供重叠的对。

对于 Python 2，您可以使用 itertools.izip 而不是 zip 来获取惰性迭代器：

import itertools

def pairwise(iterable):
    "s -> (s0, s1), (s1, s2), (s2, s3), ..."
    a, b = itertools.tee(iterable)
    next(b, None)
    return itertools.izip(a, b)

可以扩展该技术以使用 tee 函数的 n 参数生成更大的“窗口”。例如，要生成三元组，您可以使用：

def threes(iterator):
    "s -> (s0, s1, s2), (s1, s2, s3), (s2, s3, 4), ..."
    a, b, c = itertools.tee(iterator, 3)
    next(b, None)
    next(c, None)
    next(c, None)
    return zip(a, b, c)

请注意，如果其中一个迭代器比其他迭代器前进得更远，则使用 tee 可能会消耗内存。然而，在迭代器以相同速度前进的情况下，这种方法是内存高效且惯用的。

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