如何高效使用Zip并行迭代多列表并设置缺失元素默认值

一、Zip函数的基础并行迭代 #

zip()是Python里并行遍历多个可迭代对象的常用工具，每次从每个对象里取一个元素组成元组，遍历会在最短的那个可迭代对象耗尽时停止。

比如你给出的代码：

a = [1,2,4,6]
b = [[1,3],[4,7],[6,1]]
for num, (x, y) in zip(a, b):
    print(num, x, y)

运行后只会输出3行结果，因为b只有3个元素，a的第4个元素6没机会被处理。

二、解决元素遗漏：用zip_longest设置默认值 #

要保留所有元素，给缺失的并行元素设置默认值，得用itertools模块里的zip_longest函数，它会以最长的可迭代对象为准，缺失的位置用指定值填充。

示例代码（指定默认值） #

from itertools import zip_longest

a = [1,2,4,6]
b = [[1,3],[4,7],[6,1]]
# 给b的缺失元素设置默认元组(0, 0)
for num, elem in zip_longest(a, b, fillvalue=(0, 0)):
    x, y = elem
    print(num, x, y)

运行输出：

1 1 3
2 4 7
4 6 1
6 0 0

这样a的元素6就会被正常输出，对应的b位置用你设置的默认值填充。

处理默认None的情况 #

如果不指定fillvalue，默认会用None填充，这时候要注意解包时的报错问题，得加个判断：

from itertools import zip_longest

a = [1,2,4,6]
b = [[1,3],[4,7],[6,1]]
for num, elem in zip_longest(a, b):
    x, y = elem if elem is not None else (0, 0)
    print(num, x, y)

三、高效使用Zip的小提示 #

• 只需要遍历到最短对象的场景，直接用zip()就行，性能最优，代码也简洁。
• 要保留所有元素时，用zip_longest，记得根据后续处理逻辑设置合适的fillvalue，避免出现类型错误。

内容的提问来源于stack exchange，提问作者code_dominar