当使用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
我如何简化或改进它?
当前回答
您可以在输入数据上使用滑动窗口来查看下一个值，并使用哨兵来检测最后一个值。这适用于任何可迭代对象，所以你不需要事先知道它的长度。成对实现来自itertools recipes。
from itertools import tee, izip, chain
def pairwise(seq):
a,b = tee(seq)
next(b, None)
return izip(a,b)
def annotated_last(seq):
"""Returns an iterable of pairs of input item and a boolean that show if
the current item is the last item in the sequence."""
MISSING = object()
for current_item, next_item in pairwise(chain(seq, [MISSING])):
yield current_item, next_item is MISSING:
for item, is_last_item in annotated_last(data_list):
if is_last_item:
# current item is the last item
其他回答
如果你正在浏览这个列表，这对我来说也是有效的:
for j in range(0, len(Array)):
if len(Array) - j > 1:
notLast()
大多数情况下，让第一次迭代成为特殊情况比最后一次迭代更容易(也更便宜):
first = True
for data in data_list:
if first:
first = False
else:
between_items()
item()
这将适用于任何迭代对象，即使是那些没有len()的迭代对象:
file = open('/path/to/file')
for line in file:
process_line(line)
# No way of telling if this is the last line!
除此之外，我不认为有更好的解决方案，因为这取决于你想要做什么。例如，如果您正在从列表中构建字符串，那么使用str.join()自然比使用“带有特殊情况”的For循环更好。
使用相同的原理，但更紧凑:
for i, line in enumerate(data_list):
if i > 0:
between_items()
item()
看起来很眼熟，不是吗?：）
对于@ofko，以及其他真正需要找出不带len()的可迭代对象的当前值是否为最后一个值的人，你需要向前看:
def lookahead(iterable):
"""Pass through all values from the given iterable, augmented by the
information if there are more values to come after the current one
(True), or if it is the last value (False).
"""
# Get an iterator and pull the first value.
it = iter(iterable)
last = next(it)
# Run the iterator to exhaustion (starting from the second value).
for val in it:
# Report the *previous* value (more to come).
yield last, True
last = val
# Report the last value.
yield last, False
然后你可以这样使用它:
>>> for i, has_more in lookahead(range(3)):
... print(i, has_more)
0 True
1 True
2 False
你可以用下面的代码确定最后一个元素:
for i,element in enumerate(list):
if (i==len(list)-1):
print("last element is" + element)
我只是遇到了这个问题，我的通用解决方案使用迭代器:
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
我在下面分享了两个简单的方法来查找循环的结束。
方法1:
num_list = [1, 2, 3, 4]
for n in num_list:
if num_list[-1] == n:
print('this is the last iteration of the loop')
方法2:
num_list = [1, 2, 3, 4]
loop_count = len(num_list) - 1 # 3
for index, num in enumerate(num_list):
if index == loop_count:
print('this is the last iteration of the loop')
