假設我們有一個生成器函數gen()，我們不知道它是否為空。如果它是空的，我們想執行一個特殊的函數foo()，否則我們想在迭代器的每個元素上執行一個函數bar(elem)。我可以這樣做：is_empty = Truefor elem in gen(): is_empty = False bar(elem)if is_empty: foo()但這感覺不是很pythonic。還有其他方法嗎？
2 回答
喵喵時光機
TA貢獻1846條經驗 獲得超7個贊
有這個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)
您仍然需要窺視生成器，但您可以做到干凈利落。
翻翻過去那場雪
TA貢獻2065條經驗 獲得超14個贊
您無法直接判斷生成器是否為空。這是設計使然。生成器背后的一個關鍵原則是它們不會在內存中保存生成序列的所有項目。
但是你可以做這樣的事情：
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)
添加回答
舉報
0/150
提交
取消