Grouping, combinations and batches
batched needs Python 3.12 or later (the browser runs 3.13). The batches mimic splitting documents into groups for an embedding API.
itertools is the standard library's toolbox for iterators: infinite counters, chaining, slicing, grouping, combinations, running totals, and splitting into batches. Every tool is lazy and works on any iterable.
Change the code and press Run (Ctrl+Enter). Predict what will happen first, then break it on purpose and read what changes. Your edits are saved and match the lesson page.
Ready for more? This phase has 6 hands-on exercises with automatic checks.
Practice questions
Write the code in the editor, run it, then open the model answer to compare.
Use islice and count to print the first five multiples of 7 as a list.
Use accumulate to turn the daily sales [5, 3, 8, 2] into running totals and print them.
Group the words ["ant", "ape", "bee", "bat", "cow"] by first letter and print each letter with the number of words. (Hint: are the words sorted by first letter?)
Use pairwise to print the differences between neighbours of [2, 5, 9, 14].
Explain it without notes
What does itertools.groupby do, and what must you do to the data first?
Why are the itertools tools lazy, and what are the consequences?
What is the difference between permutations and combinations? Give the result of each for "abc" taken 2 at a time.
Why would you use batched in an AI application?
You call list(islice(it, 3)) twice on the same iterator it. What do you get, and why?
Expected output
a ['apple', 'avocado']
b ['banana', 'blueberry']
c ['cherry']
pairwise: [(1, 4), (4, 9), (9, 16)]
product: [('a', 1), ('a', 2), ('b', 1), ('b', 2)]
permutations: [('a', 'b'), ('a', 'c'), ('b', 'a'), ('b', 'c'), ('c', 'a'), ('c', 'b')]
combinations: [('a', 'b'), ('a', 'c'), ('a', 'd'), ('b', 'c'), ('b', 'd'), ('c', 'd')]
send to embedding API: ('d1', 'd2')
send to embedding API: ('d3', 'd4')
send to embedding API: ('d5',)