计算循环中处理的元素数的 Pythonic 方法

问题描述

使用 enumerate 适用于大于 0 的值，但在 0 处中断：

>>> def process_data(elements):
...     for cnt,value in enumerate(range(elements)):
...         pass
...     print(f'Processed {cnt + 1} elements')
... 
>>> process_data(3)
Processed 3 elements
>>> process_data(0)
Traceback (most recent call last):
  File "<stdin>",line 1,in <module>
  File "<stdin>",line 4,in process_data
UnboundLocalError: local variable 'cnt' referenced before assignment

是否有更好的 Pythonic 方法来计算循环中处理的条目数？

这当然是可行的，但是为什么要使用 enumerate：

>>> def process_data(elements):
...     processed = -1
...     for cnt,value in enumerate(range(elements)):
...         processed = cnt
...     print(f'Processed {processed + 1} elements')
... 
>>> process_data(3)
Processed 3 elements
>>> process_data(0)
Processed 0 elements

解决方法

你可以在 enumerate 循环之前给 index 变量一个初始值

count = 0
for count,obj in enumerate(objects,start=1):
    pass
print(f'Processed {count} elements')

您可以将初始值传递给 enumerate 以从 1 开始，而不是在之后将 1 添加到计数中

,

为什么不是这个？

def process_data(elements):
    processed = 0
    for processed,value in enumerate(range(elements),start=1):
        # do something
    print(f'Processed {processed} elements')