Как получить n-й элемент списка Python или значение по умолчанию, если он недоступен

Я ищу в Python эквивалент dictionary.get(key, default)для списков. Есть ли какая-то одна идиома лайнера для получения n-го элемента списка или значения по умолчанию, если оно недоступно?

Например, учитывая список myList, который я хотел бы получить myList[0], или 5, если myListэто пустой список.

Спасибо.

l[index] if index < len(l) else default

Для поддержки отрицательных индексов мы можем использовать:

l[index] if -len(l) <= index < len(l) else default

try:
   a = b[n]
except IndexError:
   a = default

Изменить: я удалил проверку TypeError - возможно, лучше позволить вызывающему обработать это.

(a[n:]+[default])[0]

Это, вероятно, лучше, чем aбольше

(a[n:n+1]+[default])[0]

Это работает, потому что если a[n:]это пустой список, еслиn => len(a)

Вот пример того, как это работает с range(5)

>>> range(5)[3:4]
[3]
>>> range(5)[4:5]
[4]
>>> range(5)[5:6]
[]
>>> range(5)[6:7]
[]

И полное выражение

>>> (range(5)[3:4]+[999])[0]
3
>>> (range(5)[4:5]+[999])[0]
4
>>> (range(5)[5:6]+[999])[0]
999
>>> (range(5)[6:7]+[999])[0]
999

Только что обнаружил, что:

next(iter(myList), 5)

iter(l)возвращает итератор myList, next()потребляет первый элемент итератора и вызывает StopIterationошибку, кроме случаев, когда вызывается со значением по умолчанию, как в данном случае, второй аргумент,5

Это работает только тогда, когда вам нужен 1-й элемент, как в вашем примере, но не в тексте вашего вопроса, поэтому ...

Кроме того, ему не нужно создавать временные списки в памяти, и он работает для любого типа итераций, даже если у него нет имени (см. Комментарий Xiong Chiamiov к ответу gruszczy)

(L[n:n+1] or [somedefault])[0]

... ищу в Python эквивалент dict.get(key, default)для списков

Есть рецепты itertools, которые делают это для общих итераций. Для удобства вы можете > pip install more_itertoolsимпортировать и эту стороннюю библиотеку, которая реализует для вас такие рецепты:

Код

import more_itertools as mit


mit.nth([1, 2, 3], 0)
# 1    

mit.nth([], 0, 5)
# 5    

Деталь

Вот реализация nthрецепта:

def nth(iterable, n, default=None):
    "Returns the nth item or a default value"
    return next(itertools.islice(iterable, n, None), default)

Мол dict.get(), этот инструмент возвращает значение по умолчанию для отсутствующих индексов. Это применимо к общим итерациям:

mit.nth((0, 1, 2), 1)                                      # tuple
# 1

mit.nth(range(3), 1)                                       # range generator (py3)
# 1

mit.nth(iter([0, 1, 2]), 1)                                # list iterator 
# 1  

Дешевое решение - действительно сделать диктант с перечислением и использовать .get()как обычно, например

 dict(enumerate(l)).get(7, my_default)

Комбинируя @ Joachim с вышеперечисленным, вы можете использовать

next(iter(my_list[index:]), default)

Примеры:

next(iter(range(10)[8:]), 11)
8
>>> next(iter(range(10)[12:]), 11)
11

Или, может быть, более ясно, но без len

my_list[index] if my_list[index:] else default

Использование Python 3.4 contextlib.suppress(exceptions)для создания getitem()метода, аналогичного getattr().

import contextlib

def getitem(iterable, index, default=None):
    """Return iterable[index] or default if IndexError is raised."""
    with contextlib.suppress(IndexError):
        return iterable[index]
    return default

Прочитав ответы, я буду использовать:

(L[n:] or [somedefault])[0]