当使用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
我如何简化或改进它?
当前回答
我只是遇到了这个问题，我的通用解决方案使用迭代器:
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
其他回答
“code between”是头尾模式的一个例子。
你有一个项目，后面是一系列(项目之间)对。您还可以将其视为(item, between)对的序列，后跟一个item。通常更简单的做法是将第一个元素作为特殊元素，而将所有其他元素作为“标准”情况。
此外，为了避免重复代码，必须提供一个函数或其他对象来包含不想重复的代码。在循环中嵌入if语句，除非有一次总是假的，这有点傻。
def item_processing( item ):
# *the common processing*
head_tail_iter = iter( someSequence )
head = next(head_tail_iter)
item_processing( head )
for item in head_tail_iter:
# *the between processing*
item_processing( item )
这更可靠，因为它更容易证明，它不会创建额外的数据结构(即列表的副本)，也不需要大量浪费if条件的执行，if条件总是假的，只有一次例外。
我将提供一种更优雅和健壮的方式，如下所示，使用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错误
我只是遇到了这个问题，我的通用解决方案使用迭代器:
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
如果要循环List， 使用枚举函数是最好的尝试之一。
for index, element in enumerate(ListObj):
# print(index, ListObj[index], len(ListObj) )
if (index != len(ListObj)-1 ):
# Do things to the element which is not the last one
else:
# Do things to the element which is the last one
如果你只是想修改data_list中的最后一个元素，那么你可以简单地使用符号:
L[-1]
然而，看起来你做的还不止这些。你的方式并没有什么问题。我甚至快速浏览了一些Django的模板标签代码，它们做的基本和你做的一样。
