Python Language itertools.takewhile


Exemple

itertools.takewhile vous permet de prendre des éléments d'une séquence jusqu'à ce qu'une condition devienne la première False .

def is_even(x):
    return x % 2 == 0


lst = [0, 2, 4, 12, 18, 13, 14, 22, 23, 44]
result = list(itertools.takewhile(is_even, lst))

print(result)
  

Cette sortie [0, 2, 4, 12, 18] .

Notez que le premier nombre qui viole le prédicat (c'est-à-dire la fonction renvoyant une valeur booléenne) is_even est 13 . Une fois que takewhile rencontre une valeur qui produit False pour le prédicat donné, elle éclate.

La sortie produite par takewhile est similaire à la sortie générée par le code ci-dessous.

def takewhile(predicate, iterable):
    for x in iterable:
        if predicate(x):
            yield x
        else:
            break

Note: La concaténation des résultats produits par takewhile et dropwhile produit l'itérable original.

result = list(itertools.takewhile(is_even, lst)) + list(itertools.dropwhile(is_even, lst))