如何扩展 Python 的 zip() 函数来处理不等长度的可迭代对象？

扩展 zip() 功能：填充到最长长度

Python 的内置 zip() 函数对来自多个可迭代对象的元素进行配对，但它将结果截断为最短可迭代的长度。如果您需要使用 None 值填充的更全面的 zip 以与最长的输入对齐，请考虑以下解决方案：

Python 3：itertools.zip_longest

在 Python 中3、itertools提供了zip_longest()函数。它扩展结果列表以匹配最长输入的长度。

import itertools
a = ['a1']
b = ['b1', 'b2', 'b3']
c = ['c1', 'c2']

list(itertools.zip_longest(a, b, c))
# [('a1', 'b1', 'c1'), (None, 'b2', 'c2'), (None, 'b3', None)]

您可以使用 fillvalue 参数指定自定义填充值：

list(itertools.zip_longest(a, b, c, fillvalue='foo'))
# [('a1', 'b1', 'c1'), ('foo', 'b2', 'c2'), ('foo', 'b3', 'foo')]

Python 2: itertools.izip_longest 或 map None

在 Python 2 中，您可以使用itertools.izip_longest （Python 2.6 中引入）或使用带有 None 的 map。

from itertools import izip_longest
a = ['a1']
b = ['b1', 'b2', 'b3']
c = ['c1', 'c2']

list(izip_longest(a, b, c))
# [('a1', 'b1', 'c1'), (None, 'b2', 'c2'), (None, 'b3', None)]

map(None, a, b, c)
# [('a1', 'b1', 'c1'), (None, 'b2', 'c2'), (None, 'b3', None)]

以上是如何扩展 Python 的 zip() 函数来处理不等长度的可迭代对象？的详细内容。更多信息请关注PHP中文网其他相关文章！