用尽的迭代器——如何处理它们？

Question

（在 Python 3.1 中） （与another question I asked 有点相关，但这个问题是关于迭代器被耗尽的问题。）

# trying to see the ratio of the max and min element in a container c
filtered = filter(lambda x : x is not None and x != 0, c)
ratio = max(filtered) / min(filtered)

我花了半个小时才意识到问题出在哪里（过滤器返回的迭代器在它到达第二个函数调用时已经耗尽）。如何以最 Pythonic / 规范的方式重写它？

另外，除了获得更多经验之外，我还能做些什么来避免此类错误？ （坦率地说，我不喜欢这种语言特性，因为这些类型的错误很容易制造，很难捕捉。）

Answer 1

您只需调用tuple(iterator) 即可将迭代器转换为元组

但是我会将该过滤器重写为列表理解，看起来像这样

# original
filtered = filter(lambda x : x is not None and x != 0, c)

# list comp
filtered = [x for x in c if x is not None and x != 0]

Answer 2

实际上，您的代码引发了一个可以防止此问题的异常！所以我猜问题是你掩盖了异常？

>>> min([])
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: min() arg is an empty sequence
>>> min(x for x in ())
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: min() arg is an empty sequence

无论如何，你也可以编写一个新函数来同时给你最小值和最大值：

def minmax( seq ):
    " returns the `(min, max)` of sequence `seq`"
    it = iter(seq)
    try:
        min = max = next(it)
    except StopIteration:
        raise ValueError('arg is an empty sequence')
    for item in it:
        if item < min:
            min = item
        elif item > max:
            max = item
    return min, max

Answer 3

实体filtered本质上是一个有状态的对象。当然，现在很明显，在其上运行 max 或 min 会改变该状态。为了避免绊倒，我想绝对清楚（对我自己来说，真的）我正在构建一些东西，而不仅仅是改造一些东西：

添加一个额外的步骤真的很有帮助：

def filtered(container):
    return filter(lambda x : x is not None and x != 0, container)

ratio = max(filtered(c)) / min(filtered(c))

您是否将filtered(...) 放在某个函数中（也许其他任何东西都不需要）或将其定义为模块级函数取决于您，但在这种情况下，我建议如果filtered ( iterator) 仅在函数中需要，将其保留在那里，直到您在其他地方需要它。

您可以做的另一件事是从中构造一个list，它将评估迭代器：

filtered_iter = filter(lambda x : x is not None and x != 0, container)
filtered = list(filtered_iter)

ratio = max(filtered) / min(filtered)

（当然，你可以直接说filtered = list(filter(...))。）

Answer 4

itertools.tee 函数可以在这里提供帮助：

import itertools

f1, f2 = itertools.tee(filtered, 2)
ratio = max(f1) / min(f2)