仅使用itertools重复可迭代(可以是无限的)中的每个元素

Question

下面有没有一个更简单的表达式，它将使用itertools，而不需要定义函数？

它必须支持任何可迭代的，包括来自生成器的无限序列。

我看了里面的食谱https://docs.python.org/3/library/itertools.html没有发现任何明显的东西。

def repeat_each(iterable, n):
"""Repeat each element in iterable n times"""
it = iter(iterable)
for element in it:
for _ in range(n):
yield element
a = itertools.cycle('abcd')
b = repeat_each(a, 4)
print(''.join(itertools.islice(b, 30)))
# Output: 'aaaabbbbccccddddaaaabbbbccccdd'

Answer 1

我不确定它是否涵盖了您心目中的所有情况，但您可以使用tee(复制迭代器n次(，然后使用zip和chainit:

from itertools import tee, chain, cycle, islice
a = cycle('abcd')
b = chain.from_iterable(zip(*tee(a, 4)))
print(''.join(islice(b, 30)))

结果：

aaaabbbbccccddddaaaabbbbccccdd