Comment obtenir le nième élément d'une liste python ou une valeur par défaut s'il n'est pas disponible

Je recherche un équivalent en python de dictionary.get(key, default)for lists. Y a-t-il un idiome de ligne pour obtenir le nième élément d'une liste ou une valeur par défaut si elle n'est pas disponible?

Par exemple, étant donné une liste myList que je voudrais obtenir myList[0], ou 5 si myListest une liste vide.

Merci.

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

Pour prendre en charge les indices négatifs, nous pouvons utiliser:

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

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

Edit: J'ai supprimé la vérification de TypeError - probablement mieux de laisser l'appelant gérer cela.

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

C'est probablement mieux en agrossissant

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

Cela fonctionne car si a[n:]est une liste vide sin => len(a)

Voici un exemple de comment cela fonctionne avec range(5)

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

Et la pleine expression

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

Je viens de découvrir que:

next(iter(myList), 5)

iter(l)renvoie un itérateur sur myList, next()consomme le premier élément de l'itérateur, et lève une StopIterationerreur sauf s'il est appelé avec une valeur par défaut, ce qui est le cas ici, le deuxième argument,5

Cela ne fonctionne que lorsque vous voulez le 1er élément, ce qui est le cas dans votre exemple, mais pas dans le texte de votre question, donc ...

De plus, il n'a pas besoin de créer de listes temporaires en mémoire et fonctionne pour tout type d'itérable, même s'il n'a pas de nom (voir le commentaire de Xiong Chiamiov sur la réponse de gruszczy)

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

... à la recherche d'un équivalent en python de dict.get(key, default)for lists

Il existe des recettes itertools qui le font pour les itérables généraux. Pour plus de commodité, vous pouvez > pip install more_itertoolset importer cette bibliothèque tierce qui implémente de telles recettes pour vous:

Code

import more_itertools as mit


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

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

Détail

Voici la mise en œuvre de la nthrecette:

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

Comme dict.get(), cet outil renvoie une valeur par défaut pour les index manquants. Elle s'applique aux itérables généraux:

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  

Une solution bon marché consiste à créer un dict avec enumerate et à utiliser .get()comme d'habitude, comme

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

En combinant @ Joachim avec ce qui précède, vous pouvez utiliser

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

Exemples:

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

Ou, peut-être plus clair, mais sans le len

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

Utiliser Python 3.4 contextlib.suppress(exceptions)pour construire une getitem()méthode similaire à 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