ジェネレーター式

はじめに

ジェネレーター式の構文

gen = (x**2 for x in range(5))
print(gen)  # generator object
print(next(gen))  # 0
print(next(gen))  # 1

メモリ効率

total = sum(x**2 for x in range(100))
print(total)

ジェネレーターの消費

gen = (x for x in range(3))
for val in gen:
    print(val)
# gen is now exhausted

関数引数としてのジェネレーター

print(sum(x**2 for x in range(10)))
print(max(len(w) for w in ['cat', 'elephant', 'ant']))

ジェネレーターの連結

evens = (x for x in range(100) if x % 2 == 0)
big_evens = (x for x in evens if x > 50)
print(list(big_evens))

リストやタプルへの変換

gen = (x**2 for x in range(5))
print(list(gen))

ジェネレーター式とリスト内包表記

import sys
lst = [x**2 for x in range(1000)]
gen = (x**2 for x in range(1000))
print(sys.getsizeof(lst), sys.getsizeof(gen))

無限ジェネレーター

import itertools
first10 = list(itertools.islice(itertools.count(0), 10))
print(first10)

ジェネレーターでの any() と all()

nums = [1, 3, 7, 2]
print(any(x > 5 for x in nums))
print(all(x > 0 for x in nums))

パイプライン

data = range(1000)
filtered = (x for x in data if x % 7 == 0)
transformed = (x * 2 for x in filtered)
print(sum(transformed))

理解度チェック

まとめ

次へ進みましょう