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

你不能直接判断发电机是否是空的。这是按设计的。生成器背后的一个关键原则是,它们不在内存中保存生成序列的所有项。

但是你可以这样做:

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)

这里有 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)

你仍然需要窥视发电机,但你可以做得很干净。