Hoàn toàn chọn các mục từ một danh sách hoặc tuple

Tôi có danh sách Python sau (cũng có thể là một tuple):

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

tôi có thể nói

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

Làm thế nào để tôi chọn ra rõ ràng các mục mà chỉ số của chúng không có mẫu cụ thể? Ví dụ, tôi muốn chọn [0,2,3]. Hoặc từ một danh sách rất lớn gồm 1000 mặt hàng, tôi muốn chọn [87, 342, 217, 998, 500]. Có một số cú pháp Python làm điều đó? Một cái gì đó trông giống như:

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

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

Tôi đã so sánh các câu trả lời với 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] )

Lưu ý rằng trong Python 3, 1st đã được thay đổi thành giống với 4th.

Một tùy chọn khác là bắt đầu với một numpy.arraycho phép lập chỉ mục thông qua một danh sách hoặc 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])

Các tuplekhông làm việc theo cách giống như đó là những lát.

Cái này thì sao:

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

Nó không được tích hợp sẵn, nhưng bạn có thể tạo một lớp con của danh sách lấy các bộ dữ liệu làm "chỉ mục" nếu bạn muốn:

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]

in ấn

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

Có thể một sự hiểu biết danh sách theo thứ tự:

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

Sản xuất:

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

Có phải đó là những gì bạn đang tìm kiếm?

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

Bạn cũng có thể tạo Listlớp riêng hỗ trợ các bộ dữ liệu làm đối số __getitem__nếu bạn muốn có thể thực hiện myList[(2,2,1,3)].

Tôi chỉ muốn chỉ ra, thậm chí cú pháp của itemgetter trông rất gọn gàng, nhưng nó hơi chậm khi thực hiện trên danh sách lớn.

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

Mục đã lấy 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))

Nhiều lát lấy 0,6225321444745759

Một giải pháp khả thi khác:

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)

như thường khi bạn có một mảng nooly boolean như mask

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

Một lambda hoạt động cho bất kỳ chuỗi hoặc np.array:

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

newseq = subseq(myseq, mask)