如何在 Python 中迭代成对或成组的列表？

列表上的成对迭代

成对迭代元素是操作列表时的常见要求。然而，Python 中的循环和列表推导式的标准并未提供成对遍历的内置解决方案。

成对实现

为了克服此限制，可以实现称为pairwise() 的自定义函数。此函数接受可迭代对象作为输入并返回元素对。

def pairwise(iterable):
    "s -> (s0, s1), (s2, s3), (s4, s5), ..."
    a = iter(iterable)
    return zip(a, a)

用法

使用此pairwise() 函数，您可以对元素进行迭代，如下所示如下：

l = [1, 2, 3, 4, 5, 6]

for x, y in pairwise(l):
    print("{} + {} = {}".format(x, y, x + y))

输出：

1 + 2 = 3
3 + 4 = 7
5 + 6 = 11

广义分组

对于需要迭代的情况任何大小的组中的元素，可以使用更通用的函数 grouped()

def grouped(iterable, n):
    "s -> (s0,s1,s2,...sn-1), (sn,sn+1,sn+2,...s2n-1), (s2n,s2n+1,s2n+2,...s3n-1), ..."
    return zip(*[iter(iterable)] * n)

用法

for x, y in grouped(l, 2):
    print("{} + {} = {}".format(x, y, x + y))

使用 Mypy 进行类型检查

对于希望执行的 Python 3 用户使用 Mypy 进行类型检查，grouped() 函数可以注释为如下：

from typing import Iterable, Tuple, TypeVar

T = TypeVar("T")

def grouped(iterable: Iterable[T], n=2) -> Iterable[Tuple[T, ...]]:
    """s -> (s0,s1,s2,...sn-1), (sn,sn+1,sn+2,...s2n-1), ..."""
    return zip(*[iter(iterable)] * n)

以上是如何在 Python 中迭代成对或成组的列表？的详细内容。更多信息请关注PHP中文网其他相关文章！