一日一技：不等长度列表的同时迭代

Posted 2020-12-21

一日一技：不等长度列表的同时迭代

有两个列表 [1,2,3]和 [‘a‘,‘b‘,‘c‘]，当我们想对他们同时迭代的时候，可以这样写：

a = [1, 2, 3]
b = [‘a‘, ‘b‘, ‘c‘]

for pair in zip(a, b):
print(pair)
运行效果如下图所示：

这种用法相信大家都已经知道了。

现在问题来了，有两个序列： [1,2,3] 和 [‘a‘,‘b‘,‘c‘,‘d‘,‘e‘]，如果我继续使用 zip对他们进行迭代，会发现 d和 e丢失了：

如果我们把 a, b交换一下位置，会发现依然没有 d, e，如下图所示：

如果我想实现的把整个 [‘a‘,‘b‘,‘c‘,‘d‘,‘e‘]都遍历完，并且 [1,2,3]缺的部分使用 NA代替，应该怎么办呢？

此时可以使用： zip_longest:

from itertools import zip_longest
a = [1, 2, 3]
b = [‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘]

for pair in zip_longest(b, a, fillvalue=‘NA‘):
print(pair)
运行效果如下图所示：

大家也可以把a，b交换一下位置，可以发现这样也是成立的。自带的zip是基于最短的列表来进行迭代，而zip_longest是以最长的序列来进行迭代。

zip_longest也可以支持多个序列，它会以最长的为准：

from itertools import zip_longest
a = [1, 2, 3]
b = [‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘]
c = [‘x‘, ‘y‘, ‘z‘, ‘w‘]

for pair in zip_longest(a, b, c, fillvalue=‘NA‘):
print(pair)
运行效果如下图所示：

kingname