Merging Iterables with Fillvalue

  • Share this:

Code introduction


The function merges multiple iterable objects. If the lengths of the iterables are not consistent, it fills in the missing parts with `fillvalue`.


Technology Stack : Built-in library

Code Type : Function

Code Difficulty : Intermediate


                
                    
def zip_longest(*args, fillvalue=None):
    iters = [iter(arg) for arg in args]
    while True:
        result = []
        for it in iters:
            try:
                result.append(next(it))
            except StopIteration:
                result.append(fillvalue)
        if len(result) == 1 and result[0] is fillvalue:
            return
        yield result