Yield, faire des générateurs comme des fonctions
Grâce a yield nous pouvons crées des générateurs complexes presque de la même manière qu'une fonction. Par exemple nous pouvons faire une fonction filtrent les nombres non premiers d'un générateur.
def premier(iterable):
for number in iterable:
if number>1:
for n in range(2, number):
if number%n==0:
break
else:
yield number
for prim in premier([2, 4, 4, 7, 8, 10, 83, 87, 373, 997]):
print(prim, end=" ") # out : 2 7 83 373 997
Nous pouvons imaginer d'autre exemple! Ils sont aussi très utiles dans le parcours complexe comme itérer dans plusieurs listes en gardant l'ordre croissant sans avoir retrié les listes ou en faire une nouvelle.
def iter_sorted_list(list_of_iterable):
iterators = [iter(iterable) for iterable in list_of_iterable]
nexts = []
ends = []
for i, iterator in enumerate(iterators):
try:
nexts.append(next(iterator))
except StopIteration:
ends.append(i)
for di, delete in enumerate(ends):
iterators.pop(delete-di)
while iterators:
i_iterator, min_value = 0, nexts[0]
for i, next_value in enumerate(nexts[1:], start=1):
if next_value<min_value:
i_iterator, min_value = i, next_value
try:
nexts[i_iterator] = next(iterators[i_iterator])
except StopIteration:
nexts.pop(i_iterator)
iterators.pop(i_iterator)
yield min_value
list_of_iterables = [[1, 3, 4],
[1, 2, 8],
[],
[10, 47, 145],
[-1, 4, 7, 57, 150],
[1,4]]
for i in iter_sorted_list(list_of_iterables):
print(i, end=" ") # out : -1 1 1 1 2 3 4 4 4 7 8 10 47 57 145 150