Python Program to Split a List Into Evenly Sized Chunks – 2022

Machine Learning Projects

Hey guys, in this blog we will see a Python Program to Split a List Into Evenly Sized Chunks.

Example 1: Using yield

def split(list_a, chunk_size):

  for i in range(0, len(list_a), chunk_size):
    yield list_a[i:i + chunk_size]

chunk_size = 2
my_list = [1,2,3,4,5,6,7,8,9]
print(list(split(my_list, chunk_size)))

Output

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

  • Using yield to print the results.
  • Chunk size is 2 so it will start like printing list_a[0:2] then list_a[2:4] then list_a[4:6] and so on…

Example 2: Using NumPy

import numpy as np

my_list = [1,2,3,4,5,6,7,8,9]
print(np.array_split(my_list, 5))

Output

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

  • Numpy has a built-in function that creates splits of an array using the array_split() method.
  • It just needs an array and a split size.

Check out our other python programming examples

Leave a Comment

Your email address will not be published. Required fields are marked *