-
Notifications
You must be signed in to change notification settings - Fork 0
/
chunked.py
42 lines (34 loc) · 1.08 KB
/
chunked.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
from itertools import islice
NOTHING = object()
def chunked(sequence, n, *, fill=NOTHING):
iterator=iter(sequence)
while iterator:
if not ( ret := list(islice(iterator, n)) ):
return
if len(ret) == n:
yield ret
else:
yield ret+[fill]*(n-len(ret)) if fill is not NOTHING else ret
def chunked_v1(sequence, n, *, fill=NOTHING):
iterator=iter(sequence)
try:
while iterator:
ret = []
for i in range(n):
ret.append(next(iterator))
yield ret
except StopIteration:
if ret:
yield ret+[fill]*(n-len(ret)) if fill is not NOTHING else ret
if __name__ == "__main__":
for chunk in chunked([1,2,3,4,5],n=2):
print(*chunk)
for chunk in chunked(range(10), 4):
print(tuple(chunk))
squares = (n**2 for n in range(6))
for chunk in chunked(squares, 3):
print(*chunk)
for chunk in chunked(range(10), 4, fill=0):
print(*chunk)
for chunk in chunked(range(4), 2, fill=0):
print(*chunk)