Expressions génératrices

Introduction

Syntaxe des expressions génératrices

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

Efficacité mémoire

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

Consommer un générateur

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

Générateur comme argument de fonction

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

Chaîner des générateurs

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))

Convertir en liste ou en tuple

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

Expression génératrice ou compréhension de liste

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))

Générateurs infinis

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

any() et all() avec des générateurs

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

Chaînes de traitement

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

Vérification rapide

Récapitulatif

Continuez