python 从迭代中生成连续n个大小的块的生成器。

Posted 2021-05-09

from itertools import islice, chain
 
 
def chunks(iterable: iter, n: int) -> iter:
    '''
   Generator that yields successive n-sized chunks from an iterable.
   Each yilded chunk is a generator of size=chunk_size ().
   That means you have to empty (iterate over) current chunk in order to get the right result
   Last chunk will be sized as is and will not be padded with any value.
   ABCDEFGH -> ABC, DEF, GH
   >>> for chunk in chunks(range(25), n=7):
   ...     print(list(chunk))
   [0, 1, 2, 3, 4, 5, 6]
   [7, 8, 9, 10, 11, 12, 13]
   [14, 15, 16, 17, 18, 19, 20]
   [21, 22, 23, 24]
   '''
    iterator = iter(iterable)
    for first in iterator:
        yield chain([first], islice(iterator, n - 1))
 
if __name__ == "__main__":
    import doctest
    doctest.testmod(verbose=True)