split_iterable

pyhelpers.ops.split_iterable(iterable, chunk_size)

Split a list into (evenly sized) chunks.

See also [OPS-SI-1].

Parameters

• iterable (Iterable) – iterable object

• chunk_size (int) – length of a chunk

Returns

a sequence of equally-sized chunks from iterable

Return type

Generator[Iterable]

Examples:

>>> from pyhelpers.ops import split_iterable
>>> import pandas

>>> iterable_1 = list(range(0, 10))
>>> iterable_1
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

>>> iterable_1_ = split_iterable(iterable_1, chunk_size=3)
>>> type(iterable_1_)
generator

>>> for dat in iterable_1_:
...     print(list(dat))
[0, 1, 2]
[3, 4, 5]
[6, 7, 8]
[9]

>>> iterable_2 = pandas.Series(range(0, 20))
>>> iterable_2
0      0
1      1
2      2
3      3
4      4
5      5
6      6
7      7
8      8
9      9
10    10
11    11
12    12
13    13
14    14
15    15
16    16
17    17
18    18
19    19
dtype: int64

>>> iterable_2_ = split_iterable(iterable_2, chunk_size=5)

>>> for dat in iterable_2_:
...     print(list(dat))
[0, 1, 2, 3, 4]
[5, 6, 7, 8, 9]
[10, 11, 12, 13, 14]
[15, 16, 17, 18, 19]