koaning / embetter

just a bunch of useful embeddings

Home Page:https://koaning.github.io/embetter/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add `batched` utility.

koaning opened this issue · comments

From itertools:

def batched(iterable, n):
    "Batch data into tuples of length n. The last batch may be shorter."
    # batched('ABCDEFG', 3) --> ABC DEF G
    if n < 1:
        raise ValueError('n must be at least one')
    it = iter(iterable)
    while batch := tuple(islice(it, n)):
        yield batch