Як отримати 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()споживає перший елемент ітератора і піднімає aStopIteration помилку, за винятком випадків, коли викликається зі значенням за замовчуванням, що є випадком тут, другий аргумент,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's з вищезазначеним, ви могли б скористатися

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's 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