คีย์เวิร์ด yield

บทนำ

yield คืออะไร

def count_up(n):
    for i in range(n):
        yield i
for x in count_up(3):
    print(x)

ฟังก์ชันตัวสร้างเทียบกับฟังก์ชันปกติ

def gen():
    print('start')
    yield 1
    print('middle')
    yield 2
g = gen()  # no print yet
print(next(g))  # 'start', then 1

ตัวสร้างแบบไม่สิ้นสุด

def naturals(n=0):
    while True:
        yield n
        n += 1
import itertools
print(list(itertools.islice(naturals(), 5)))

yield from

def chain(*iters):
    for it in iters:
        yield from it
print(list(chain([1,2], [3,4], [5])))

การส่งค่าด้วย send()

def accumulator():
    total = 0
    while True:
        n = yield total
        if n is None: break
        total += n
g = accumulator()
next(g)  # prime
print(g.send(10))
print(g.send(5))

สถานะของตัวสร้าง

def stateful():
    seen = set()
    while True:
        item = yield
        seen.add(item)
        yield len(seen)
g = stateful()
next(g); g.send('a'); print(next(g)); g.send('b'); print(next(g))

การปิดตัวสร้าง

def resource_gen():
    try:
        while True:
            yield
    finally:
        print('cleaned up')
g = resource_gen()
next(g)
g.close()

StopIteration และการคืนค่า

def gen_with_return():
    yield 1
    yield 2
    return 'done'
g = gen_with_return()
try:
    while True: print(next(g))
except StopIteration as e:
    print('returned:', e.value)

ตัวสร้างฟีโบนัชชี

def fibonacci():
    a, b = 0, 1
    while True:
        yield a
        a, b = b, a + b
import itertools
print(list(itertools.islice(fibonacci(), 10)))

ตัวสร้างเทียบกับคลาสที่มี __iter__

# Generator is simpler than:
# class Counter:
#     def __iter__(self): return self
#     def __next__(self): ...
print('generators simplify iterators')

ตรวจสอบความเข้าใจ

ทบทวน

เรียนต่อไป