当使用for循环迭代时，我如何处理输入的最后一个元素?特别是，如果有代码应该只出现在元素之间(而不是在最后一个元素之后)，我该如何构造代码?
目前，我写的代码是这样的:
for i, data in enumerate(data_list):
code_that_is_done_for_every_element
if i != len(data_list) - 1:
code_that_is_done_between_elements
我如何简化或改进它?
当前回答
我发现在循环表达式之前定义循环值很方便。对于这个方框示例，匹配循环中的值，或者其他可能需要它的地方。
numberofboxes = 1411
for j in range(1,numberofboxes):
if j != numberofboxes - 1:
print ("},")
else:
print("}")
其他回答
谷歌让我想到了这个老问题，我想我可以用不同的方法来解决这个问题。
这里的大多数答案将处理一个正确的处理for循环控件，但如果data_list是可销毁的，我建议你从列表中弹出项目，直到你得到一个空列表:
while True:
element = element_list.pop(0)
do_this_for_all_elements()
if not element:
do_this_only_for_last_element()
break
do_this_for_all_elements_but_last()
如果不需要对最后一个元素做任何操作，甚至可以使用while len(element_list)。我发现这个解决方案比处理next()更优雅。
将最后一项的特殊处理延迟到循环之后。
>>> for i in (1, 2, 3):
... pass
...
>>> i
3
除了向上数，你也可以向下数:
nrToProcess = len(list)
for s in list:
s.doStuff()
nrToProcess -= 1
if nrToProcess==0: # this is the last one
s.doSpecialStuff()
假设input是一个迭代器，下面是使用itertools中的tee和izip的方法:
from itertools import tee, izip
items, between = tee(input_iterator, 2) # Input must be an iterator.
first = items.next()
do_to_every_item(first) # All "do to every" operations done to first item go here.
for i, b in izip(items, between):
do_between_items(b) # All "between" operations go here.
do_to_every_item(i) # All "do to every" operations go here.
演示:
>>> def do_every(x): print "E", x
...
>>> def do_between(x): print "B", x
...
>>> test_input = iter(range(5))
>>>
>>> from itertools import tee, izip
>>>
>>> items, between = tee(test_input, 2)
>>> first = items.next()
>>> do_every(first)
E 0
>>> for i,b in izip(items, between):
... do_between(b)
... do_every(i)
...
B 0
E 1
B 1
E 2
B 2
E 3
B 3
E 4
>>>
我只是遇到了这个问题，我的通用解决方案使用迭代器:
from typing import TypeVar, Iterable
E = TypeVar('E')
def metait(i: Iterable[E]) -> Iterable[tuple[E, bool, bool]]:
first = True
previous = None
for elem in i:
if previous:
yield previous, first, False
first = False
previous = elem
if previous:
yield previous, first, True
您将收到一个元组，其中包含第一项和最后一项的原始元素和标志。它可以用于每个可迭代对象:
d = {'a': (1,2,3), 'b': (4,5,6), 'c': (7,8,9)}
for (k,v), is_first, is_last in metait(d.items()):
print(f'{k}: {v} {is_first} {is_last}')
这将给你:
a: (1, 2, 3) True False
b: (4, 5, 6) False False
c: (7, 8, 9) False True
