python基础——zip函数和zip_longest函数

参考内置函数——python3.8.1文档

zip函数

1、创建一个聚合了来自每个可迭代对象中的元素的迭代器，返回一个元组的迭代器。
2、其中的第 i 个元组包含来自每个参数序列或可迭代对象的第 i 个元素。
3、 当所输入可迭代对象中最短的一个被耗尽时，迭代器将停止迭代。
4、当只有一个可迭代对象参数时，它将返回一个单元组的迭代器。
5、 不带参数时，它将返回一个空迭代器。

a = [1,2,3]
print(list(zip(a)))  # 只有一个可迭代参数
# 输出：[(1,), (2,), (3,)]

a = [1,2,3]
b = [4,5,6]
print(list(zip(a,b)))  # 两个可迭代参数长度相同时
# 输出：[(1, 4), (2, 5), (3, 6)]

a = [1,2,3,4]
b = [5,6,7]
print(list(zip(a,b))) # 两个可迭代参数长度不同时，按最短的组合输出
# 输出：[(1, 5), (2, 6), (3, 7)]

a = [[1,2,3], [4,5,6]]
b = [[7,8,9], [10,11,12]]
print(list(zip(a,b)))
# 输出：[([1, 2, 3], [7, 8, 9]), ([4, 5, 6], [10, 11, 12])]

zip_longest函数

zip_longest与zip函数唯一不同的是如果两个可迭代参数长度不同时，按最长的输出，长度不足的用None进行代替

from itertools import zip_longest

a = [1,2,3,4]
b = [5,6,7]
print(list(zip_longest(a,b)))
# 输出：[(1, 5), (2, 6), (3, 7), (4, None)]