最近在看流暢的python，在看第14章節(jié)的itertools模塊,對其itertools中的相關(guān)函數(shù)實現(xiàn)的邏輯的實現(xiàn)

其中在zip_longest(it_obj1, ..., it_objN, fillvalue=None)時，其函數(shù)實現(xiàn)的功能和內(nèi)置zip函數(shù)大致相同(實現(xiàn)一一對應(yīng))，

不過內(nèi)置的zip函數(shù)是已元素最少對象為基準(zhǔn),而zip_longest函數(shù)是已元素最多對象為基準(zhǔn),使用fillvalue的值來填充

以下是自己總結(jié)此函數(shù)的大致實現(xiàn)方法,和官方方法不同:

思路大致如此: 找出元素個數(shù)最多 ==>算出元素個數(shù)差值==>填充差值個元素到各個對象

def zip_longest(*it, **kwargs):  its = {k: len(k) for k in it}   # 這里我是用字典把參數(shù)對象和參數(shù)的元素個數(shù)結(jié)果作為一個字典  max_num = max(its.values())    # 確定迭代對象元素最大值   result = []            #   fillvalue = kwargs.get('fillvalue', None) # 元素較少的填充值  for x in range(max_num):     # 已最大次為基準(zhǔn)循環(huán)    result = []    for it in its:        # 循環(huán)所有迭代對象,以便進(jìn)行填充數(shù)據(jù)      element = list(it)    # 將      if len(it) < max_num:   # 如果迭代對象的元素個數(shù)比最大值小,則要填充數(shù)據(jù)        for i in range(max_num - len(it)): # 此為要填充數(shù)據(jù)的個數(shù)          element.append(fillvalue)    # 填充操作,完成后所有的迭代對象的元素個數(shù)都為一致          result.append(element[x])        # 生成一一對應(yīng)結(jié)果，存放到list中    yield tuple(result)             # 將結(jié)果輸出

測試其結(jié)果:

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

結(jié)果為:

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