blob: d7b8305322c8d7d9ef847bd626b3e4077e04e6f7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
|
from itertools import *
# Python 3.12:
if "batched" not in globals():
# https://docs.python.org/3/library/itertools.html#itertools.batched
def batched(iterable, n):
# 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
|
