Question

让我们假设我们有一个生成器函数gen()，我们不知道它是否为空。

如果为空，我们想执行一个特殊的函数foo()，否则我们想对迭代器bar(elem)的每个元素执行一个函数。

我可以这样：

is_empty = True
for elem in gen():
    is_empty = False
    bar(elem)
if is_empty: foo()

但是那感觉不是很pythonic。还有其他方法吗？

Answer 1

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

但是您可以执行以下操作：

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)

Answer 2

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

您仍然需要窥视发生器，但是您可以干净地做它。

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

2 个答案: