Как по-разному относиться к последнему элементу в списке в Python? - PullRequest
46 голосов
/ 12 марта 2010

Мне нужно выполнить специальную операцию для последнего элемента в списке. Есть ли лучший способ, чем этот?

array = [1,2,3,4,5] 
for i, val in enumerate(array): 
  if (i+1) == len(array): 
    // Process for the last element 
  else: 
    // Process for the other element 

Ответы [ 5 ]

55 голосов
/ 12 марта 2010
for item in list[:-1]:
    print "Not last: ", item
print "Last: ", list[-1]

Если вы не хотите делать копию списка, вы можете создать простой генератор:

# itr is short for "iterable" and can be any sequence, iterator, or generator

def notlast(itr):
    itr = iter(itr)  # ensure we have an iterator
    prev = itr.next()
    for item in itr:
        yield prev
        prev = item

# lst is short for "list" and does not shadow the built-in list()
# 'L' is also commonly used for a random list name
lst = range(4)
for x in notlast(lst):
    print "Not last: ", x
print "Last: ", lst[-1]

Другое определение для notlast:

import itertools
notlast = lambda lst:itertools.islice(lst, 0, len(lst)-1)
34 голосов
/ 12 марта 2010

Если ваша последовательность не очень длинная, вы можете просто нарезать ее:

for val in array[:-1]:
  do_something(val)
else:
  do_something_else(array[-1])
6 голосов
/ 12 марта 2010

с помощью itertools

>>> from itertools import repeat, chain,izip
>>> for val,special in izip(array, chain(repeat(False,len(array)-1),[True])):
...     print val, special
... 
1 False
2 False
3 False
4 False
5 True

Версия ответа Лиори для работы на любой итерации (не требует len() или нарезки)

def last_flagged(seq):
    seq = iter(seq)
    a = next(seq)
    for b in seq:
        yield a, False
        a = b
    yield a, True        

mylist = [1,2,3,4,5]
for item,is_last in last_flagged(mylist):
    if is_last:
        print "Last: ", item
    else:
        print "Not last: ", item
0 голосов
/ 25 октября 2016

Простой способ с условием if:

for item in list:
    print "Not last: ", item
    if list.index(item) == len(list)-1:
        print "Last: ", item
0 голосов
/ 17 октября 2014
for i in items:
  if i == items[-1]:
    print 'The last item is: '+i
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...