Python如何檢查迭代器工具鏈中是否已達到最後一個元素？

Question

for elt in itertools.chain.from_iterable(node): 

if elt is the last element: 
    do statement 

我該如何做到這一點

Answer 1

當循環結束時，elt變量不會超出範圍，並且仍然保留循環給出的最後一個值。所以你可以把代碼放在循環結尾，並在elt變量上運行。這不是非常漂亮，但是Python的範圍規則也不是很好。

這個問題（謝謝，cvondrick）唯一的問題是循環可能永遠不會執行，這將意味着elt不存在 - 我們會得到一個NameError。所以，充分的方式做這將是大致爲：

del elt # not necessary if we haven't use elt before, but just in case 
for elt in itertools.chain.from_iterable(node): 
    do_stuff_to_each(elt) 
try: 
    do_stuff_to_last(elt) 
except NameError: # no last elt to do stuff to 
    pass 

Answer 2

你本身不可以。您需要存儲當前項目，推進迭代器，並捕獲StopIteration異常。然後你需要以某種方式表明你有最後一個項目。

Answer 3

您可以通過使用iter.next()手動推進迭代器在while循環，然後捕捉StopIteration例外做到這一點：

>>> from itertools import chain 
>>> it = chain([1,2,3],[4,5,6],[7,8,9]) 
>>> while True: 
...  try: 
...   elem = it.next() 
...  except StopIteration: 
...   print "Last element was:", elem, "... do something special now" 
...   break 
...  print "Got element:", elem 
...  
... 
Got element: 1 
Got element: 2 
Got element: 3 
Got element: 4 
Got element: 5 
Got element: 6 
Got element: 7 
Got element: 8 
Got element: 9 
Last element was: 9 ... do something special now 
>>> 

Answer 4

我做了什麼像這樣：

rng = len(mlist) 
for i in range(rng): 
    foo = mlist[i] 
    foo.do_something_for_every_item_regardless() 
    if i == rng - 1: #since we go from 0 to rng-1 
     foo.last_item_only_operation()