Python: Split an iterable and generate iterables specified number of times

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

Write a Python program to split an iterable and generate iterables a specified number of times.

Sample Solution:

Python Code:

import itertools as it
def tee_data(iter, n):
    return it.tee(iter, n)
#List
result = tee_data(['A','B','C','D'], 5)
print("Generate iterables specified number of times:")
for i in result:
    print(list(i))

#String
result = tee_data("Python itertools", 4)
print("\nGenerate iterables specified number of times:")
for i in result:
    print(list(i))

Sample Output:

Generate iterables specified number of times:
['A', 'B', 'C', 'D']
['A', 'B', 'C', 'D']
['A', 'B', 'C', 'D']
['A', 'B', 'C', 'D']
['A', 'B', 'C', 'D']

Generate iterables specified number of times:
['P', 'y', 't', 'h', 'o', 'n', ' ', 'i', 't', 'e', 'r', 't', 'o', 'o', 'l', 's']
['P', 'y', 't', 'h', 'o', 'n', ' ', 'i', 't', 'e', 'r', 't', 'o', 'o', 'l', 's']
['P', 'y', 't', 'h', 'o', 'n', ' ', 'i', 't', 'e', 'r', 't', 'o', 'o', 'l', 's']
['P', 'y', 't', 'h', 'o', 'n', ' ', 'i', 't', 'e', 'r', 't', 'o', 'o', 'l', 's']

Python Code Editor:

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

Previous: Write a Python program to create an iterator that returns consecutive keys and groups from an iterable.
Next: Write a Python program to create an iterator to get specified number of permutations of elements.