CODEWITHSUNDEEP
python
prosseek
prosseek

Reputation: 190729

How to treat the last element in list differently in Python?

I need to do some special operation for the last element in a list. Is there any better way than this?

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

Upvotes: 55

Views: 91910

Answers (7)

Isaac To
Isaac To

Reputation: 141

Use more_itertools:

import more_itertools

array = [1,2,3,4,5]
peekable = more_itertools.peekable(array)
last = object()
for val in peekable:
    if peekable.peek(last) is last: 
        print('last', val)
    else: 
        print(val)

Gives:

1
2
3
4
last 5

Upvotes: 0

Pedro M Duarte
Pedro M Duarte

Reputation: 28093

If your logic is never breaking out of the loop then a for...else construct might work:

In [1]: count = 0
   ...: for i in [1, 2, 3]:
   ...:     count +=1
   ...:     print("any item:", i)
   ...: else:
   ...:     if count:
   ...:         print("last item: ", i)
   ...:
any item: 1
any item: 2
any item: 3
last item:  3

You need the count variable just in case the iterable is empty, otherwise the variable i won't be defined.

Upvotes: 0

Max Dy
Max Dy

Reputation: 307

Simple way with an if condition:

for item in list:
    print "Not last: ", item
    if list.index(item) == len(list)-1:
        print "Last: ", item

Upvotes: -3

ahmad88me
ahmad88me

Reputation: 1

for i in items:
  if i == items[-1]:
    print 'The last item is: '+i

Upvotes: -4

liori
liori

Reputation: 42267

for item in list[:-1]:
    print "Not last: ", item
print "Last: ", list[-1]

If you don't want to make a copy of list, you can make a simple generator:

# 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]

Another definition for notlast:

import itertools
notlast = lambda lst:itertools.islice(lst, 0, len(lst)-1)

Upvotes: 68

John La Rooy
John La Rooy

Reputation: 304137

using 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

Version of liori's answer to work on any iterable (doesn't require len() or slicing)

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

Upvotes: 9

Ignacio Vazquez-Abrams
Ignacio Vazquez-Abrams

Reputation: 798566

If your sequence isn't terribly long then you can just slice it:

for val in array[:-1]:
  do_something(val)
else:
  do_something_else(array[-1])

Upvotes: 42

Related Questions