Python: Chunk a given list into smaller lists of a specified size

Last update on December 21 2024 13:11:00 (UTC/GMT +8 hours)

Chunk List by Size

Write a Python program to chunk a given list into smaller lists of a specified size.

• Use list() and range() to create a list of the desired size.
• Use map() on the list and fill it with splices of the given list.
• Finally, return the created list.

Sample Solution:

Python Code:

# Import the 'ceil' function from the 'math' module.
from math import ceil

# Define a function 'chunk_list' that takes a list 'lst' and an integer 'size' as input.
def chunk_list(lst, size):
    # Use list comprehensions to create a list of chunks, where each chunk has 'size' elements.
    # The 'range' function is used to generate the indices for splitting the list.
    return list(map(lambda x: lst[x * size:x * size + size],  # Lambda function to get each chunk.
            list(range(ceil(len(lst) / size)))))  # Calculate the number of chunks needed and generate indices.

# Call the 'chunk_list' function with an example list and chunk size.
print(chunk_list([1, 2, 3, 4, 5, 6, 7, 8], 3))

Sample Output:

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

Flowchart:

Python Code Editor:

Previous: Write a Python program to split values into two groups, based on the result of the given filter list.
Next: Write a Python program to chunk a given list into n smaller lists.