不,这不安全。迭代器不是序列。这就是发生的事情foo()
使用生成器的函数,生成器本身就是一个迭代器:
>>> foo(x for x in range(10))
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 1, in foo
ValueError: max() arg is an empty sequence
这是因为当 after 之后迭代器已经结束了sum()
完成了它的工作,所以max()
没有从中得到任何额外的物品。一般来说,不可能重置迭代器以使其再次循环。为了foo()
函数要正确支持迭代器,必须重写它,以便迭代器仅循环一次,例如通过保存来自iterator
进入暂时的tuple
or list
:
def foo(iterator):
iterable = list(iterator)
return sum(iterable) / max(iterable)
or, if iterator
产生大量的物品,通过小心使用a来处理它for
loop:
def foo(iterator):
# allows iterables like lists or tuples to be passed as arguments
iterator = iter(iterator)
try:
max_ = next(iterator)
sum_ = max_
except StopIteration:
# iterator yields no items, return some default value
return 0.0
for element in iterator:
sum_ += element
max_ = max(max_, element)
return sum_ / max_
这将产生正确的结果:
>>> foo(x for x in range(10))
5.0