Level 4Exercise #99

Batch an iterator

Group an iterator into tuples of size n; allow a shorter final batch.

What you’ll practice

This Level 4 exercise focuses on batched in the Itertools curriculum. Solve the challenge prompt above using clear, idiomatic Python and the relevant language or standard-library tools.

Relevant Python reference: itertools.batched.

Sample Test Cases

Example #1

Input:batches(range(5), batch_size=2)

Output:[(0,1),(2,3),(4,)]

Reference Solution

Reveal Reference Solution

def batches(iterable, batch_size: int):
    from itertools import islice
    it = iter(iterable)
    out = []
    while batch := list(islice(it, batch_size)):
        out.append(batch)
    return out

Pro Tips & Keyboard Shortcuts

Color Theme

Ctrl + K then T

Open VS Code Color Theme Quick Pick to select from 20 dark and light themes.

Editor SettingsCtrl + K

Open practice settings drawer to toggle line numbers, indenting, font size & hints.

Search MenuCtrl + /

Expand sidebar menu, focus search bar, and highlight search text instantly.

Focus Code Editor

Esc or Ctrl + `

Instantly highlight and focus code editor from anywhere, restoring cursor right where you left off.

Normal ViewEsc

Collapse sidebar and close all popups or settings drawers for clean focus view.

Batch an iterator

What you’ll practice

Sample Test Cases

Further Reading

Reference Solution