Como obter o enésimo elemento de uma lista python ou um padrão, se não estiver disponível

Estou procurando um equivalente em python de dictionary.get(key, default)para listas. Existe algum idioma de liner para obter o enésimo elemento de uma lista ou um valor padrão, se não estiver disponível?

Por exemplo, dada uma lista myList que eu gostaria de obter myList[0], ou 5 se myListfor uma lista vazia.

Obrigado.

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

Para suportar índices negativos, podemos usar:

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

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

Edit: Eu removi a verificação para TypeError - provavelmente melhor deixar o chamador lidar com isso.

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

Provavelmente é melhor à medida que aaumenta

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

Isso funciona porque if a[n:]é uma lista vazia sen => len(a)

Aqui está um exemplo de como isso funciona com range(5)

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

E a expressão completa

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

Acabei de descobrir que:

next(iter(myList), 5)

iter(l)retorna um iterador myList, next()consome o primeiro elemento do iterador e gera um StopIterationerro, exceto se chamado com um valor padrão, que é o caso aqui, o segundo argumento,5

Isso só funciona quando você deseja o primeiro elemento, como é o caso no seu exemplo, mas não no texto da sua pergunta, então ...

Além disso, ele não precisa criar listas temporárias na memória e funciona para qualquer tipo de iterável, mesmo que não tenha um nome (consulte o comentário de Xiong Chiamiov na resposta de gruszczy)

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

... procurando um equivalente em python de dict.get(key, default)para listas

Existem receitas de ferramentas que fazem isso para iteráveis ​​gerais. Por conveniência, você pode > pip install more_itertoolsimportar esta biblioteca de terceiros que implementa essas receitas para você:

Código

import more_itertools as mit


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

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

Detalhe

Aqui está a implementação da nthreceita:

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

Assim dict.get(), essa ferramenta retorna um padrão para os índices ausentes. Aplica-se a iterables gerais:

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  

Uma solução barata é realmente fazer um ditado com enumerar e usar .get()como de costume, como

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

Combinando o @ Joachim's com o acima, você pode usar

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

Exemplos:

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

Ou, talvez mais claro, mas sem o len

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

Usando o Python 3.4 contextlib.suppress(exceptions)para criar um getitem()método semelhante ao 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