チェーンのpython

Question

で同じ発電機の繰り返し次の簡単なジェネレータ考えてみましょう：

def simple_gen(): 
    for number in range(5): 
     yield number ** 2 

を私はチェーンにユーザーitertools.repeatとitertools.chainに喜んn回発生装置自体に。明確にするために同じの次（非発電機）の例を検討：

array = [1,2,3,4] 
repetitions = itertools.repeat(array ,2) 
list(itertools.chain.from_iterable(repetitions)) -> [1,2,3,4,1,2,3,4] 

を私が同じことをしたいが、配列の代わりに自分の発電機（simple_gen）を使用します。もちろん、itertools.repeatが同じオブジェクトを繰り返すため、単純な置換が機能しないため、ジェネレータの次の繰り返しが使い果たされます。

itertoolsモジュールを使用してこれを達成する方法のいくつかのアイデアはありますか？

ジェネレータをリストまたは別のコンテナにキャストしたくありません。

Answer 1

あなたは最初のリストに発電機の出力を変換することができます：

repeat(list(simple_gen()), 2) 

あなたはは、そうでない場合は、発電機出力を繰り返すことはできません。代わりのchain.from_iterable(repeat(callable(), count))のPython 2との使用のため

from itertools import repeat 

def recreate(callable, count=None): 
    for c in repeat(callable, count): 
     yield from c() 

のPython 3で

と

from itertools import repeat 

def recreate(callable, count=None): 
    for c in repeat(callable, count): 
     for val in c(): 
      yield val 

：せいぜいあなたは 回の発電 数番号を再作成することができます。ジェネレータ関数はと呼ばれていない であることに注意してください。代わりに関数オブジェクト自体を渡してください。

デモ：

>>> from itertools import repeat 
>>> def simple_gen(): 
...  for number in range(5): 
...   yield number ** 2 
... 
>>> def recreate(callable, count=None): 
...  for c in repeat(callable, count): 
...   for val in c(): 
...    yield val 
... 
>>> list(recreate(simple_gen, 2)) 
[0, 1, 4, 9, 16, 0, 1, 4, 9, 16] 

Answer 2

どうデコレータについて：

from itertools import repeat 

class repeat_gen: 
    def __init__(self, count): 
     self.count = count 

    def __call__(self, gen): 
     for it in repeat(gen, self.count): 
      yield from it() 

@repeat_gen(3) 
def simple_gen(): 
    for number in range(5): 
     yield number ** 2 

print(list(simple_gen)) 

注すぎitertools.repeat作品せずに、次のことを。 simple_genは汎用関数としてデコレータに渡されるため、複数回呼び出すことができます。

class repeat_gen: 
    def __init__(self, count): 
     self.count = count 

    def __call__(self, gen): 
     self.gen = gen 
     def repeat_gen(): 
      for i in range(self.count): 
       yield from self.gen() 
     return repeat_gen 

@repeat_gen(3) 
def simple_gen(): 
    for number in range(5): 
     yield number ** 2 

print(list(simple_gen()))