pythonのジェネレータ関数　基礎的な使い方がわかりません

pythonのジェネレータの使い方を知りたいです

def gen(maxnum):
    base=0
    i=0
    while i < maxnum:
        o=(yield base+i)
        
        if o is not None:
            base = o
        else:
            i+=1
            
g=gen(10)
#1
print(next(g))
#2
print(next(g))
g.send(10)
#3
print(next(g))
#4
print(next(g))
g.send(0)
#5
print(next(g))
#6
print(next(g))
#7
print(next(g))

実行結果

試したこと

pythonの絵本から持ってきたコードです

print(next(g))がどのように処理され値が呼び出されるのかいまいちわかりません。

def gen(maxnum):
    base = 0
    i = 0
    while i < maxnum:
        o=(yield base + i)
        
        if o is not None:
            base = o
        else:
            i += 1

私なりには #1 print(next(g))　に呼び出しされ　g=gen(10)　が処理され上記defの関数の式が開始されて、
gen(maxnum)のところに(10)が代入され
while i < maxnum(10):となり、while文が処理されると考えていいのでしょうか？

while i < maxnum:
        o=(yield base + i)
        
        if o is not None:
            base = o
        else:
            i += 1

ここで処理では　o=(yield base + i)　で中断し呼び出し元にに戻ると記載されているのですが
o= base + iが戻り値になるのですか？

if o is not None:
            base = o
        else:
            i += 1

次に#2 print(next(g))が上記のコードから再開されるのでしょうか？
if o is not None:　の意味はわかるのですがo変数がNoneになる場合とならない場合の処理を教えてください。

g.send(10)

send()メゾットはどこに値を送るのでしょうか？

補足情報（FW/ツールのバージョンなど）

情報不足ですみません。

quickquip

2020/11/04 01:09

書いてあるコードはなんですか? 「こんな感じだと思ったけど動かないコード」ですか? そうであれば「どう動くことを期待したコードか?」「どのように動かないのか? エラーになるのか、意図しない動作になるのか?」の説明が必要です。「自分には意味は分からないけれどどこかから拾ってきた動くコード」ですか? そうであれば「どこに書かれているどういうコードなのか?」「実際に自分で動かしてみたのか? どういう結果になったのか?」の説明が必要です。その上で「いったいどこが疑問点なのか?」を書いてください。それと、今のままではコードが読めません。 https://teratail.com/help/question-tips#questionTips3-5-1 を参考に編集からコードを選択して <code> ボタンでコードの前後行に ``` をいれて読めるようにしてください。

nanata_umu

2020/11/04 12:44

アドバイスありがとうございます。情報が足りないようであれば随時追加します。説明足らずで申し訳ございません。

行動規範の内容に同意します

回答2件

ベストアンサー

こんな感じです。

ジェネレータも普通の関数ど同様に処理が行なわれます。関数はreturnで値を返すと、その関数は終ってしまいますが、yieldで値を返すと、returnと違っても関数が終了せず生きていて、nextを呼び出すことでで続きができるのです。
また、呼び出し側がジェネレータのsend()で、yieldの返り値を渡すことができます。

処理を追うとこんな感じです。

g=gen(0) でジェネレータが生成されます。通常の関数とちがって、ここでは実行結果の返り値ではなく、生成されたジェネレータがgに入ります。

next(g) で、yieldまで実行され、base+i が呼び出し元に返ります。
最初は base=0 で i=0なので、0が返ります。
さらにnext(g)が呼ばれると、yieldの次の行から再開します。この時、yieldの返り値は、send()を呼んでいないので、None におり、if文はelse側が実行され、ループが回って、yield base+i (base=0, i=1) で 1 が返ります。

g.send(10) を実行すると、関数内のyieldの返り値にsendの結果が渡されます。ということで o に 10 が入ることになります。
ここで next(g) を呼ぶと、if文はoの値がNoneではないので、てbaseの値がo すなわち sendで渡された10に書き換えます。
yield base+i は base=10 で i=2 なので、12を返します。

以降はご自分で追いかけてみてください。

投稿2020/11/04 13:47