检测生成器函数是否为空，否则对其进行迭代

Question

假设我们有一个生成器函数gen()，我们不知道它是否为空。如果它是空的，我们想执行一个特殊的函数foo()，否则我们想在迭代器的每个元素上执行一个函数bar(elem)。我可以这样做：is_empty = Truefor elem in gen():    is_empty = False    bar(elem)if is_empty: foo()但这感觉不是很pythonic。还有其他方法吗？

Answer 1

有这个peek功能：

import itertools

def peek(iterable):

    try:

        first = next(iterable)

    except StopIteration:

        return None

    return first, itertools.chain([first], iterable)

res = peek(gen())

if not res:

    foo()

else:

    for elem in res[1]:

        bar(elem)

您仍然需要窥视生成器，但您可以做到干净利落。

Answer 2

您无法直接判断生成器是否为空。这是设计使然。生成器背后的一个关键原则是它们不会在内存中保存生成序列的所有项目。

但是你可以做这样的事情：

from itertools import chain

def check_first_and_iterate(iterable):

    try:

        first = next(iterable)

        for item in chain([first], item):

            bar(item)

    except StopIteration:

        foo()

check_first_and_iterate(iterable)