Явно выберите элементы из списка или кортежа Python

Question

У меня есть следующий список Python (также может быть кортежем):

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

Я могу сказать

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

Как явно выбрать элементы, индексы которых не имеют конкретныхузоры?Например, я хочу выбрать [0,2,3].Или из очень большого списка из 1000 предметов, я хочу выбрать [87, 342, 217, 998, 500].Есть ли какой-то синтаксис Python, который делает это?Нечто похожее:

>>> myBigList[87, 342, 217, 998, 500]

Answer 1

list( myBigList[i] for i in [87, 342, 217, 998, 500] )

---

Я сравнил ответы с python 2.5.2:

• 19,7 usec: [ 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] )

Обратите внимание, что в Python 3 1-й был изменен, чтобы быть таким же, как 4-й.

---

Другой вариант - начать с numpy.array, который позволяет индексировать по списку, или 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])

tuple работает не так, как срезы.

Answer 2

Что по этому поводу:

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

Answer 3

Он не является встроенным, но вы можете создать подкласс списка, который принимает кортежи в качестве «индексов», если вы хотите:

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]

печать

foo
['baaz', 'mumble']
['bar', 'quux']

Answer 4

>>> map(myList.__getitem__, (2,2,1,3))
('baz', 'baz', 'bar', 'quux')

Вы также можете создать свой собственный класс List, который поддерживает кортежи в качестве аргументов для __getitem__, если вы хотите иметь возможность myList[(2,2,1,3)].

Answer 5

Может быть, список в порядке:

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

Производит:

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

Это то, что вы ищете?

Answer 6

Другое возможное решение:

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)

Answer 7

Я просто хочу отметить, что даже синтаксис itemgetter выглядит очень аккуратно, но он довольно медленный, когда выполняет большой список.срез занял 0,6225321444745759

Answer 8

как часто, когда у вас есть логический массив Numpy, например mask

[mylist[i] for i in np.arange(len(mask), dtype=int)[mask]]

Лямбда, которая работает для любой последовательности или np.array:

subseq = lambda myseq, mask : [myseq[i] for i in np.arange(len(mask), dtype=int)[mask]]

newseq = subseq(myseq, mask)