イテレータのpythonリピートリスト要素

Question

リストの要素を特定の時間繰り返すイテレータを作成する方法はありますか？例えば、リストが与えられます。以下のリストを生成することができますitertools.repeatlist(color, 7)の形でイテレータを作成する方法は

color = ['r', 'g', 'b'] 

ありますか？

color_list = ['r', 'g', 'b', 'r', 'g', 'b', 'r'] 

Answer 1

あなたのrepeatlist()機能を構築するためにitertools.islice()と一緒itertools.cycle()を使用することができます。

from itertools import cycle, islice 

def repeatlist(it, count): 
    return islice(cycle(it), count) 

これは、新しいイテレータを返します。リストオブジェクトが必要な場合は、list()を呼び出してください。

デモ：

>>> from itertools import cycle, islice 
>>> def repeatlist(it, count): 
...  return islice(cycle(it), count) 
... 
>>> color = ['r', 'g', 'b'] 
>>> list(repeatlist(color, 7)) 
['r', 'g', 'b', 'r', 'g', 'b', 'r'] 

Answer 2

試してみてください。

def repeat_list(l, n): 
    for i in range(n): yield l[i%len(l)]