เลือกรายการจากรายการหรือทูเพิลอย่างชัดเจน

ฉันมีรายการ 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] )

ฉันเปรียบเทียบคำตอบกับ 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ไม่ทำงานวิธีการเช่นเดียวกับผู้ที่มีชิ้น

แล้วสิ่งนี้:

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

มันไม่ได้อยู่ในตัว แต่คุณสามารถสร้างคลาสย่อยของรายการที่ใช้ tuples เป็น "ดัชนี" ได้หากคุณต้องการ:

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ชั้นที่สนับสนุน tuples เป็นข้อโต้แย้งที่จะถ้าคุณต้องการที่จะสามารถที่จะทำ__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)

เช่นเดียวกับเมื่อคุณมีอาร์เรย์ตัวเลขบูลีนเช่น 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)