Python Program to Split a List Into Evenly Sized Chunks

Example 1: Using yield

Output

[[1, 2], [3, 4], [5, 6], [7, 8], [9]]

In the above example, we have defined a function to split the list.

• Using a for loop and range() method, iterate from 0 to the length of the list with the size of chunk as the step.
• Return the chunks using yield. list_a[i:i+chunk_size] gives each chunk. For example, when i = 0, the items included in the chunk are i to i + chunk_size which is 0 to (0 + 2)th index. In the next iteration, the items included are 2 to 2 + 2 = 4.

Learn more about yield at Python Generators.

You can do the same thing using list compression as below.

Output

[[1, 2], [3, 4], [5, 6], [7, 8], [9]]

Learn more about list comprehension at Python List Comprehension.

Example 2: Using numpy

Output

[array([1, 2]), array([3, 4]), array([5, 6]), array([7, 8]), array([9])]

array_split() is a numpy method that splits a list into equal sized chunks. Here, the number of chunks is 5.

Note: You need to install numpy on your system.