当使用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
我如何简化或改进它?
当前回答
我在下面分享了两个简单的方法来查找循环的结束。
方法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')
其他回答
我想到的一个简单的解决办法是:
for i in MyList:
# Check if 'i' is the last element in the list
if i == MyList[-1]:
# Do something different for the last
else:
# Do something for all other elements
第二个同样简单的解决方法是使用计数器:
# Count the no. of elements in the list
ListLength = len(MyList)
# Initialize a counter
count = 0
for i in MyList:
# increment counter
count += 1
# Check if 'i' is the last element in the list
# by using the counter
if count == ListLength:
# Do something different for the last
else:
# Do something for all other elements
你的方法没有错，除非你有10万个循环，想要保存10万个“if”语句。在这种情况下，你可以这样做:
iterable = [1,2,3] # Your date
iterator = iter(iterable) # get the data iterator
try : # wrap all in a try / except
while 1 :
item = iterator.next()
print item # put the "for loop" code here
except StopIteration, e : # make the process on the last element here
print item
输出:
1
2
3
3
但说真的，在你的情况下，我觉得这太过分了。
在任何情况下，你可能会更幸运的切片:
for item in iterable[:-1] :
print item
print "last :", iterable[-1]
#outputs
1
2
last : 3
或者只是:
for item in iterable :
print item
print iterable[-1]
#outputs
1
2
3
last : 3
最终，一个KISS的方式来做你的事情，这将适用于任何迭代对象，包括那些没有__len__的迭代对象:
item = ''
for item in iterable :
print item
print item
Ouputs:
1
2
3
3
如果我这样做的话，对我来说很简单。
我将提供一种更优雅和健壮的方式，如下所示，使用unpacking:
def mark_last(iterable):
try:
*init, last = iterable
except ValueError: # if iterable is empty
return
for e in init:
yield e, True
yield last, False
测试:
for a, b in mark_last([1, 2, 3]):
print(a, b)
结果是:
1真正的 2真 3错误
是否不可能遍历除最后一个元素以外的所有元素，并在循环之外处理最后一个元素?毕竟，创建循环的目的与循环遍历的所有元素类似;如果一个元素需要一些特殊的东西，它就不应该在循环中。
(另请参阅这个问题:循环中的最后一个元素是否值得单独处理)
编辑:因为这个问题更多的是关于“中间”，所以要么第一个元素是特殊的，因为它没有前一个元素，要么最后一个元素是特殊的，因为它没有后继元素。
我们可以用for-else来实现
cities = [
'Jakarta',
'Surabaya',
'Semarang'
]
for city in cities[:-1]:
print(city)
else:
print(' '.join(cities[-1].upper()))
输出:
Jakarta
Surabaya
S E M A R A N G
这个想法是我们只使用for-else循环直到n-1索引，然后在for耗尽后，我们使用[-1]直接访问最后一个索引。
