python中itertools模块zip_longest函数详解

最近在看流畅的python，在看第14章节的itertools模块,对其itertools中的相关函数实现的逻辑的实

最近在看流畅的python，在看第14章节的itertools模块,对其itertools中的相关函数实现的逻辑的实现

其中在zip_longest(it_obj1, ..., it_objN, fillvalue=None)时，其函数实现的功能和内置zip函数大致相同(实现一一对应)，

不过内置的zip函数是已元素最少对象为基准,而zip_longest函数是已元素最多对象为基准,使用fillvalue的值来填充

以下是自己总结此函数的大致实现方法,和官方方法不同:

思路大致如此: 找出元素个数最多 ==>算出元素个数差值==>填充差值个元素到各个对象

def zip_longest(*it, **kwargs):
 its = {k: len(k) for k in it} # 这里我是用字典把参数对象和参数的元素个数结果作为一个字典
 max_num = max(its.values()) # 确定迭代对象元素最大值 
 result = [] # 
 fillvalue = kwargs.get('fillvalue', None) # 元素较少的填充值
 for x in range(max_num): # 已最大次为基准循环
 result = []
 for it in its: # 循环所有迭代对象,以便进行填充数据
 element = list(it) # 将
 if len(it) < max_num: # 如果迭代对象的元素个数比最大值小,则要填充数据
 for i in range(max_num - len(it)): # 此为要填充数据的个数
 element.append(fillvalue) # 填充操作,完成后所有的迭代对象的元素个数都为一致 
 result.append(element[x]) # 生成一一对应结果，存放到list中
 yield tuple(result) # 将结果输出

测试其结果:

res = zip_longest('abc', '12')
for x in res:
 print(x)

结果为:

('a', '1') ('b', '2') ('c', None)

python zip_longest python itertools模块 Python 迭代器模块 itertools