sendメソッドについて

Question

# -*- coding: utf-8 -*-

def gen(step):
	val = 0
	while True:
		val = val + step
		step = yield val

g = gen(3)
g.__next__()
g.send(10)
g.send(5)

説明欄には「再開待ちのジェネレータに、値を送出します。send()メソッドで指定した値は、yield式の値となります。」と書かれています。
プログラムの結果は最初に引数に与えた値とsend()メソッドの引数に指定した値とを加算して値を返しています。
これはyield valをstepに代入していなければsend()メソッドは反映されず、3が加算されて値が返ります。
この説明の「send()メソッドで指定した値は、yield式の値となります。」というのはsend()メソッドに10を与えたら、10を返すという意味ではないのでしょうか。

Comments

[argius]

すみません、説明欄というのはどこの説明欄を指しているのですか？

[hogeshi]

言葉足らずですみません、私が勉強しているパーフェクトpythonという参考書のことです。

Answer 1

だいたいこのような流れで処理が行われていると考えてみてください。

send()が呼び出される前は、yieldで再開待ち
  ↓
send(10)により、stepに10が代入される(step = yield valのところ)
  ↓
whileループの先頭に戻り val = val + step が評価される (val=13)
  ↓
yieldにより、呼び出し元に 13 が返され、再びyieldで再開待ち

...

ただ、元のコードでは、stepがどのように作用しているかが分かりにくいと思いました。

このように書き換えてみると、gen()、g.__next__()、g.send(x)が呼ばれたときにどのように動作しているかが分かりやすくなると思います。

• 書き換えたコード

# -*- coding: utf-8 -*-

def gen(init_val):
    print("init_val=", init_val)
    val = init_val
    step = 0
    while True:
        print("val=", val)
        print("step=", step)
        val = val + step
        step = yield val

print("---gen(3)呼び出し---")
g = gen(3)

print("---g.__next__()呼び出し---")
print( g.__next__() )
print("---")

print("---g.send(10)呼び出し---")
print( g.send(10) )
print("---")

print("---g.send(5)呼び出し---")
print( g.send(5) )
print("---")

• 実行結果

---gen(3)呼び出し---
---g.__next__()呼び出し---
init_val= 3
val= 3
step= 0
3
---
---g.send(10)呼び出し---
val= 3
step= 10
13
---
---g.send(5)呼び出し---
val= 13
step= 5
18
---

...

参考リンク:

ジェネレータの使い方 - Qiita
http://qiita.com/Kodaira_/items/32b1ef860f59df80eedb

Comments

[hogeshi]

感覚がつかめました
ありがとう。