python中itertools模块zip_longest函数详解

itertools.zip_longest

itertools模块包含创建高效迭代器的函数，这些函数的返回值不是list，而是iterator（可迭代对象）,可以用各种方式对数据执行循环操作.

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)
参考链接：https://www.jb51.net/article/141848.htm