Getting a default value on index out of range in Python

Question:

a = ['123', '2', 4]
b = a[4] or 'sss'
print b

I want to get a default value when the list index is out of range (here: 'sss').

How can I do this?

Asked By: zjm1126

||

Answers:

In the Python spirit of “ask for forgiveness, not permission”, here’s one way:

try:
    b = a[4]
except IndexError:
    b = 'sss'
Answered By: Thomas
try:
    b = a[4]
except IndexError:
    b = 'sss'

A cleaner way (only works if you’re using a dict):

b = a.get(4,"sss") # exact same thing as above

Here’s another way you might like (again, only for dicts):

b = a.setdefault(4,"sss") # if a[4] exists, returns that, otherwise sets a[4] to "sss" and returns "sss"
Answered By: Vlad the Impala

Using try/catch?

try:
    b=a[4]
except IndexError:
    b='sss'
Answered By: zoli2k

In the non-Python spirit of “ask for permission, not forgiveness”, here’s another way:

b = a[4] if len(a) > 4 else 'sss'
Answered By: Thomas

You could also define a little helper function for these cases:

def default(x, e, y):
    try:
        return x()
    except e:
        return y

It returns the return value of the function x, unless it raised an exception of type e; in that case, it returns the value y. Usage:

b = default(lambda: a[4], IndexError, 'sss')

Edit: Made it catch only one specified type of exception.

Suggestions for improvement are still welcome!

Answered By: Thomas

I’m all for asking permission (i.e. I don’t like the tryexcept method). However, the code gets a lot cleaner when it’s encapsulated in a method:

def get_at(array, index, default):
    if index < 0: index += len(array)
    if index < 0: raise IndexError('list index out of range')
    return array[index] if index < len(a) else default

b = get_at(a, 4, 'sss')
Answered By: Konrad Rudolph

Since this is a top google hit, it’s probably also worth mentioning that the standard “collections” package has a “defaultdict” which provides a more flexible solution to this problem.

You can do neat things, for example:

twodee = collections.defaultdict(dict)
twodee["the horizontal"]["the vertical"] = "we control"

Read more: http://docs.python.org/2/library/collections.html

Answered By: Jonathan Mayer

You could create your own list-class:

class MyList(list):
    def get(self, index, default=None):
        return self[index] if -len(self) <= index < len(self) else default

You can use it like this:

>>> l = MyList(['a', 'b', 'c'])
>>> l.get(1)
'b'
>>> l.get(9, 'no')
'no'
Answered By: gecco

For a common case where you want the first element, you can do

next(iter([1, 2, 3]), None)

I use this to “unwrap” a list, possibly after filtering it.

next((x for x in [1, 3, 5] if x % 2 == 0), None)

or

cur.execute("SELECT field FROM table")
next(cur.fetchone(), None)
Answered By: Pepijn

In the Python spirit of beautiful is better than ugly

Code golf method, using slice and unpacking (not sure if this was valid 4 years ago, but it is in python 2.7 + 3.3)

b, = a[4:5] or ['sss']

Nicer than a wrapper function or try-catch IMHO, but intimidating for beginners.

Using slicing without unpacking:

b = a[4] if a[4:] else 'sss'

or, if you have to do this often, and don’t mind making a dictionary

d = dict(enumerate(a))
b = d.get(4, 'sss')
Answered By: 00500005

another way:

b = (a[4:]+['sss'])[0]
Answered By: maep

If you are looking for a maintainable way of getting default values on the index operator I found the following useful:

If you override operator.getitem from the operator module to add an optional default parameter you get identical behaviour to the original while maintaining backwards compatibility.

def getitem(iterable, index, default=None):
  import operator
  try:
    return operator.getitem(iterable, index)
  except IndexError:
    return default
Answered By: Alexander Oh

If you are looking for a quick hack for reducing the code length characterwise, you can try this.

a=['123','2',4]
a.append('sss') #Default value
n=5 #Index you want to access
max_index=len(a)-1
b=a[min(max_index, n)]
print(b)

But this trick is only useful when you no longer want further modification to the list

Answered By: Vikas Kumar
Categories: questions Tags: , , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.