假設(shè)我們有一個生成器函數(shù)gen()，我們不知道它是否為空。如果它是空的，我們想執(zhí)行一個特殊的函數(shù)foo()，否則我們想在迭代器的每個元素上執(zhí)行一個函數(shù)bar(elem)。我可以這樣做：is_empty = Truefor elem in gen(): is_empty = False bar(elem)if is_empty: foo()但這感覺不是很pythonic。還有其他方法嗎？
2 回答
喵喵時光機(jī)
TA貢獻(xiàn)1846條經(jīng)驗(yàn) 獲得超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貢獻(xiàn)2065條經(jīng)驗(yàn) 獲得超14個贊
您無法直接判斷生成器是否為空。這是設(shè)計(jì)使然。生成器背后的一個關(guān)鍵原則是它們不會在內(nèi)存中保存生成序列的所有項(xiàng)目。
但是你可以做這樣的事情：
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)
添加回答
舉報(bào)
0/150
提交
取消