Python: Split a given list into specified sized chunks using itertools module

Last update on December 21 2024 07:31:05 (UTC/GMT +8 hours)

Write a Python program to split a given list into specified sized chunks using the itertools module.

Sample Solution:

Python Code:

from itertools import islice
def split_list(lst, n):
    lst = iter(lst)
    result = iter(lambda: tuple(islice(lst, n)), ())
    return list(result)

nums = [12,45,23,67,78,90,45,32,100,76,38,62,73,29,83]
print("Original list:")
print(nums)
n = 3
print("\nSplit the said list into equal size",n)
print(split_list(nums,n))
n = 4
print("\nSplit the said list into equal size",n)
print(split_list(nums,n))
n = 5
print("\nSplit the said list into equal size",n)
print(split_list(nums,n))

Sample Output:

Original list:
[12, 45, 23, 67, 78, 90, 45, 32, 100, 76, 38, 62, 73, 29, 83]

Split the said list into equal size 3
[(12, 45, 23), (67, 78, 90), (45, 32, 100), (76, 38, 62), (73, 29, 83)]

Split the said list into equal size 4
[(12, 45, 23, 67), (78, 90, 45, 32), (100, 76, 38, 62), (73, 29, 83)]

Split the said list into equal size 5
[(12, 45, 23, 67, 78), (90, 45, 32, 100, 76), (38, 62, 73, 29, 83)]

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Python program to get the index of the first element, which is greater than a specified element using itertools module.