Implementation of zip_longest with Fillvalue

  • Share this:

Code introduction


The function utilizes the built-in `zip_longest` to merge multiple iterable objects. When one iterator is exhausted, it uses a `fillvalue` to fill the remaining iterators until all iterators are exhausted.


Technology Stack : zip_longest

Code Type : Function

Code Difficulty : Intermediate


                
                    
def zip_longest(*args, fillvalue=0):
    iters = [iter(arg) for arg in args]
    while True:
        result = []
        for iter_ in iters:
            try:
                result.append(next(iter_))
            except StopIteration:
                result.append(fillvalue)
        if not any(result):  # if all iterators are exhausted
            break
        yield tuple(result)                
              
Tags: