Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2026-05-23 19:01:53 +00:00
Code Issues Projects Releases Packages Wiki Activity

[3.11] GH-98363: Update batched() recipe in docs to match 3.12 (#100323)

Browse source
This commit is contained in:
Raymond Hettinger 2022-12-17 19:10:03 -06:00 • committed by GitHub
parent bf0a334c4e
commit 09186676cf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

22
Doc/library/itertools.rst
View file

@ -881,12 +881,12 @@ which incur interpreter overhead.
raise ValueError('Expected fill, strict, or ignore')
def batched(iterable, n):
"Batch data into lists of length n. The last batch may be shorter."
"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 := list(islice(it, n))):
while (batch := tuple(islice(it, n))):
yield batch
def triplewise(iterable):
@ -1248,25 +1248,25 @@ which incur interpreter overhead.
[('a', 'b', 'c'), ('d', 'e', 'f')]
>>> list(batched('ABCDEFG', 3))
[['A', 'B', 'C'], ['D', 'E', 'F'], ['G']]
[('A', 'B', 'C'), ('D', 'E', 'F'), ('G',)]
>>> list(batched('ABCDEF', 3))
[['A', 'B', 'C'], ['D', 'E', 'F']]
[('A', 'B', 'C'), ('D', 'E', 'F')]
>>> list(batched('ABCDE', 3))
[['A', 'B', 'C'], ['D', 'E']]
[('A', 'B', 'C'), ('D', 'E')]
>>> list(batched('ABCD', 3))
[['A', 'B', 'C'], ['D']]
[('A', 'B', 'C'), ('D',)]
>>> list(batched('ABC', 3))
[['A', 'B', 'C']]
[('A', 'B', 'C')]
>>> list(batched('AB', 3))
[['A', 'B']]
[('A', 'B')]
>>> list(batched('A', 3))
[['A']]
[('A',)]
>>> list(batched('', 3))
[]
>>> list(batched('ABCDEFG', 2))
[['A', 'B'], ['C', 'D'], ['E', 'F'], ['G']]
[('A', 'B'), ('C', 'D'), ('E', 'F'), ('G',)]
>>> list(batched('ABCDEFG', 1))
[['A'], ['B'], ['C'], ['D'], ['E'], ['F'], ['G']]
[('A',), ('B',), ('C',), ('D',), ('E',), ('F',), ('G',)]
>>> s = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
>>> all(list(flatten(batched(s[:n], 5))) == list(s[:n]) for n in range(len(s)))
True