Come ottenere l'ennesimo elemento di un elenco Python o un valore predefinito se non disponibile

Sto cercando un equivalente in Python dictionary.get(key, default)per le liste. Esiste un idioma di linea per ottenere l'ennesimo elemento di un elenco o un valore predefinito se non disponibile?

Ad esempio, dato un elenco myList che vorrei ottenere myList[0], o 5 se myListè un elenco vuoto.

Grazie.

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

Per supportare gli indici negativi possiamo usare:

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

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

Modifica: ho rimosso il controllo per TypeError - probabilmente è meglio lasciare che sia il chiamante a gestirlo.

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

Questo probabilmente è meglio perché adiventa più grande

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

Questo funziona perché if a[n:]è un elenco vuoto ifn => len(a)

Ecco un esempio di come funziona range(5)

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

E la piena espressione

>>> (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

Ho appena scoperto che:

next(iter(myList), 5)

iter(l)restituisce un iteratore attivo myList, next()consuma il primo elemento dell'iteratore e genera un StopIterationerrore, tranne se chiamato con un valore predefinito, che è il caso qui, il secondo argomento,5

Funziona solo quando vuoi il primo elemento, come nel tuo esempio, ma non nel testo della tua domanda, quindi ...

Inoltre, non è necessario creare elenchi temporanei in memoria e funziona per qualsiasi tipo di iterabile, anche se non ha un nome (vedere il commento di Xiong Chiamiov sulla risposta di gruszczy)

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

... alla ricerca di un equivalente in Python dict.get(key, default)per le liste

Esistono ricette itertools che lo fanno per iterabili generali. Per comodità, puoi > pip install more_itertoolse importare questa libreria di terze parti che implementa tali ricette per te:

Codice

import more_itertools as mit


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

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

Dettaglio

Ecco l'implementazione della nthricetta:

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

Ad esempio dict.get(), questo strumento restituisce un valore predefinito per gli indici mancanti. Si applica agli iterabili generali:

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  

Una soluzione economica è davvero fare un dict con enumerare e utilizzare .get()come al solito, come

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

Combinando @ Joachim con quanto sopra, potresti usare

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

Esempi:

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

O, forse più chiaro, ma senza il len

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

Usare Python 3.4 contextlib.suppress(exceptions)per costruire un getitem()metodo simile a 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