Pythonのハイブリッドジェネレータ/ファンクション

Question

Pythonでは、ループ内の値を生成するジェネレータが、そのループの最終値が返される通常の関数として代わりに呼び出される可能性はありますか？私は引数としてフラグを設定して、そのフラグに応じて返すか返すかを選択しました。しかし、関数内にキーワードyieldが存在するだけで、それが自動的にジェネレータに変換され、Pythonはジェネレータにreturn文があると不満を持ちます。

ここでは、そのような機能の例：

def function(generator=True): 
    a = 0 
    for i in range(10): 
     a = i 
     if generator: 
      yield a 
    if not generator: 
     return a 

いくつかのケースでは、私はちょうど（例えば、最適化のための残差の関数としてそれを使用。）最終的な結果を必要とするときにいる間、このような機能は私のために有用であろう他のケース私は、各反復後に増分結果が必要です（例えば、ロボットの差動モデルを使用して、新しい新しい速度コマンドでロボットの姿勢を更新するなど）。今のところ私は2つの機能を持っています.1つは歩留まりがあり、もう1つは戻ります。その2つを組み合わせることは可能ですか？

Answer 1

これはまだジェネレータであり、リターンを呼び出しています。 とにかくジェネレータ/レギュラーの両方の機能を混在させたくありません。

とにかく結果をループする必要がある場合に備えて、イテレータの上に何かをラップすることができます。

あなたがやりたいことができ、簡単なコード：

last = None 
for last in function(): pass 

今、最後の希望の値を保持し、あなたのコードでそれを使用することができます。

発電機内部のリターンは Python 3.3 に添加し、そしてそれは、呼び出すとStopIteration（値）と等価です

return expr in a generator causes StopIteration(expr) to be raised upon exit from the generator.

In a generator, the statement

return value is semantically equivalent to

raise StopIteration(value) except that, as currently, the exception cannot be caught by except clauses within the returning generator.

Answer 2

あなたはまだその単一の値を取得するには、それを反復処理する必要がありますが、あなたは再びyieldを使用することができますこれを達成するためにreturnの代わりに。

>>> def function(generator=True): 
...  a = 0 
...  for i in range(10): 
...   a = i 
...   if generator: 
...    yield a 
...  if not generator: 
...   yield a 
... 
>>> a = function() 
>>> print(*a) 
0 1 2 3 4 5 6 7 8 9 
>>> a = function(0) 
>>> print(a) 
<generator object function at 0x0000000001603240> 
>>> print(*a) 
9 
>>> a = function(0) 

ジェネレータ内部returnは、Python 2でSyntaxErrorあるが、パイソン3. yieldとreturnを交換しない有する2および3