Skip to content

batchify

Split a list into batches and optionally display a progress bar.

Parameters

  • X (list[str])

    A list of items to be batched.

  • batch_size (int)

    The number of items in each batch.

  • desc (str) – defaults to ``

    A description to display in the progress bar.

  • tqdm_bar (bool) – defaults to True

    Whether to display a progress bar using tqdm.

Examples

>>> items = ["a", "b", "c", "d", "e", "f"]
>>> batches = list(batchify(items, batch_size=2))
>>> for batch in batches:
...     print(batch)
['a', 'b']
['c', 'd']
['e', 'f']