Splitting a List into chunks in Python

Question

Suppose that I wish to chop up a list in python into equal-length pieces, is there an elegant way to do this? I don't want to try out the ordinary way of using an extra list and keeping a counter variable to cut the list at the right indices.

Answer 1

In python the most elegant way to do this would be to use a generator function that will keep slicing the list into another list of given size. For example:

def chunk_list(data, chunk_size):

  for i in range(0, len(data), chunk_size):

    yield data[i : i + n]

data = [x for x in range(1, 51)] # Data to chunk

for chunk in chunk_list(data, 5):

  print(chunk)

The generator function allows the caller to pause it's execution and return some value, in our case we are returning a small chunk of a list using the yield keyword.