Wie kann die Gesamtzahl beim Iterieren eines Generators korrekt berechnet werden?

Question

Mir wurde immer gesagt, dass die integrierte enumerate verwendet werden kann, wenn Sie Anzahl zählen und gleichzeitig durchlaufen müssen.

Zum Beispiel ist dies ein gemeinsames Idiom in Python (Funktion gen einen Generator zurückgibt):

for index, item in enumerate(gen()): 
    # Do something with item 

# get the numbers consumed from generator 
print("% number of items processed" % (index+1,)) 

Aber wenn der Generator gibt nichts zurück? z.B. enumerate(range(0)) ist die index Variable nicht definiert.

Wir können index Variable vor der for-Schleife definieren, aber gibt es noch mehr Pythonic-Lösung, die ich nicht bewusst war?

Answer 1

enumerate(gen(), start=1) wird machen index Start von 1 zu zählen, die Notwendigkeit für index+1 zu entfernen. Ansonsten denke ich, was du hast ist schon Pythonic.

index = 0 
for index, item in enumerate(gen(), start=1): 
    # Do something with item 

print("%d number of items processed" % (index,)) 

Answer 2

Wenn Sie beabsichtigen, die index wieder zu verwenden, dann initialisieren es einfach über Ihre for Schleife wie Sie gesagt haben. Es gibt nichts "Un-Pythonic" darüber.

Wenn Sie nicht sicher sind, ob index wird definiert werden, können Sie immer versuchen, verwenden/außer:

>>> from __future__ import print_function # for Python 2 compatibility 
>>> import sys 
>>> 
>>> for i, j in enumerate(range(0)): 
... pass 
... 
>>> try: 
... print(i) 
... except NameError: 
... print('i was not defined', file=sys.stderr) 
... 
i was not defined