Comment vérifier si un générateur est vide avant de l'itérer ?

Détermination du vide du générateur avant l'itération

Lorsque vous travaillez avec des générateurs, il peut être avantageux de vérifier si le générateur est vide au préalable. Cela évite les traitements inutiles et réduit la complexité du code.

Pour y parvenir, une approche consiste à utiliser la fonction peek, qui tente de récupérer le premier élément du générateur. Si une exception StopIteration est levée, cela signifie un générateur vide. Voici une implémentation :

<code class="python">def peek(iterable):
    try:
        first = next(iterable)
    except StopIteration:
        return None
    return first, itertools.chain([first], iterable)</code>

Dans cette fonction :

• next(iterable) récupère le premier élément du générateur ou lève une exception StopIteration s'il est vide.
• Le résultat est stocké en premier. S'il est vide, first est None.
• Pour préserver l'état du générateur, l'itérable d'origine est enveloppé dans un nouveau à l'aide de itertools.chain([first], iterable).

Vous pouvez utiliser cette fonction comme suit :

<code class="python">res = peek(mysequence)
if res is None:
    # Generator is empty
elif res[0] is None:
    # Generator is not empty but first element is None
else:
    # Generator is not empty with non-None first element</code>

Cette approche détermine efficacement le vide des générateurs avant l'itération, offrant un moyen pratique d'optimiser votre code et de gérer les générateurs vides avec élégance.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!