web-dev-qa-db-fra.com

Sélection explicite d'éléments dans une liste ou un tuple Python

J'ai la liste Python suivante (peut aussi être un tuple):

myList = ['foo', 'bar', 'baz', 'quux']

Je peux dire

>>> myList[0:3]
['foo', 'bar', 'baz']
>>> myList[::2]
['foo', 'baz']
>>> myList[1::2]
['bar', 'quux']

Comment choisir explicitement des éléments dont les index n'ont pas de modèle spécifique? Par exemple, je veux sélectionner [0,2,3]. Ou parmi une très grande liste de 1000 articles, je veux sélectionner [87, 342, 217, 998, 500]. Y at-il une syntaxe Python qui fait ça? Quelque chose qui ressemble à:

>>> myBigList[87, 342, 217, 998, 500]
pythonlistselectindexingtuples
86
Kit
list( myBigList[i] for i in [87, 342, 217, 998, 500] )

J'ai comparé les réponses avec Python 2.5.2:

  • 19.7 utilisateurs: [ myBigList[i] for i in [87, 342, 217, 998, 500] ]

  • 20.6 usec: map(myBigList.__getitem__, (87, 342, 217, 998, 500))

  • 22.7 usec: itemgetter(87, 342, 217, 998, 500)(myBigList)

  • 24.6 usec: list( myBigList[i] for i in [87, 342, 217, 998, 500] )

Notez que dans Python 3, le 1er a été changé pour être le même que le 4ème.

Une autre option serait de commencer avec un numpy.array qui permet l'indexation via une liste ou un numpy.array:

>>> import numpy
>>> myBigList = numpy.array(range(1000))
>>> myBigList[(87, 342, 217, 998, 500)]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: invalid index
>>> myBigList[[87, 342, 217, 998, 500]]
array([ 87, 342, 217, 998, 500])
>>> myBigList[numpy.array([87, 342, 217, 998, 500])]
array([ 87, 342, 217, 998, 500])

La Tuple ne fonctionne pas de la même manière que ce sont des tranches.

118
Dan D.

Et ça:

from operator import itemgetter
itemgetter(0,2,3)(myList)
('foo', 'baz', 'quux')
34
Marcin

Il n'est pas intégré, mais vous pouvez créer une sous-classe de liste qui prend des n-uplets comme "index" si vous voulez:

class MyList(list):

    def __getitem__(self, index):
        if isinstance(index, Tuple):
            return [self[i] for i in index]
        return super(MyList, self).__getitem__(index)


seq = MyList("foo bar baaz quux mumble".split())
print seq[0]
print seq[2,4]
print seq[1::2]

impression

foo
['baaz', 'mumble']
['bar', 'quux']
9
Matt Anderson
>>> map(myList.__getitem__, (2,2,1,3))
('baz', 'baz', 'bar', 'quux')

Vous pouvez également créer votre propre classe List qui prend en charge les n-uplets comme arguments de __getitem__ si vous voulez pouvoir exécuter myList[(2,2,1,3)].

5
ninjagecko

Peut-être qu'une liste de compréhension est en ordre:

L = ['a', 'b', 'c', 'd', 'e', 'f']
print [ L[index] for index in [1,3,5] ]

Produit:

['b', 'd', 'f']

Est-ce ce que vous cherchez?

4
Dan Witkowski

Je tiens juste à souligner que même la syntaxe de itemgetter a l’air très soignée, mais elle est un peu lente lorsqu’elle est effectuée sur une grande liste.

import timeit
from operator import itemgetter
start=timeit.default_timer()
for i in range(1000000):
    itemgetter(0,2,3)(myList)
print ("Itemgetter took ", (timeit.default_timer()-start))

Itemgetter a pris 1.065209062149279

start=timeit.default_timer()
for i in range(1000000):
    myList[0],myList[2],myList[3]
print ("Multiple slice took ", (timeit.default_timer()-start))

Plusieurs tranches ont pris 0.6225321444745759

1
Wendao Liu

Une autre solution possible:

sek=[]
L=[1,2,3,4,5,6,7,8,9,0]
for i in [2, 4, 7, 0, 3]:
   a=[L[i]]
   sek=sek+a
print (sek)
0
fdante