목록 또는 튜플에서 명시 적으로 항목 선택

다음 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]

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

나는 대답을 파이썬 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에서는 1st가 4th와 동일하게 변경되었습니다.

또 다른 옵션은 numpy.array목록 또는 a를 통해 인덱싱을 허용 하는로 시작하는 것입니다 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사람들은 조각이 같은 방식으로 작동하지 않습니다.

이것에 대해 :

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

기본 제공되지는 않지만 원하는 경우 튜플을 "인덱스"로 사용하는 목록의 하위 클래스를 만들 수 있습니다.

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']

목록 이해가 순서대로있을 수 있습니다.

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

생성 :

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

그게 당신이 찾고있는 것입니까?

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

당신이 할 수 있기를 원한다면 List튜플을 인수로 지원하는 자신의 클래스를 만들 수도 있습니다 .__getitem__myList[(2,2,1,3)]

지적하고 싶은 것은 itemgetter의 구문조차도 정말 깔끔해 보이지만 큰 목록에서 수행 할 때는 다소 느립니다.

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가 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))

다중 슬라이스는 0.6225321444745759를 사용했습니다.

또 다른 가능한 해결책 :

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)

종종 부울 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)