ฉันจำเป็นต้องเลือกองค์ประกอบบางอย่างจากรายการที่กำหนดรู้ดัชนีของพวกเขา สมมติว่าฉันต้องการสร้างรายการใหม่ซึ่งมีองค์ประกอบที่มีดัชนี 1, 2, 5 จากรายการที่ระบุ [-2, 1, 5, 3, 8, 5, 6] สิ่งที่ฉันทำคือ:

a = [-2,1,5,3,8,5,6]
b = [1,2,5]
c = [ a[i] for i in b]

มีวิธีที่ดีกว่าที่จะทำหรือไม่ อะไรอย่าง c = a [b]?

คุณสามารถใช้operator.itemgetter:

from operator import itemgetter 
a = [-2, 1, 5, 3, 8, 5, 6]
b = [1, 2, 5]
print(itemgetter(*b)(a))
# Result:
(1, 5, 5)

หรือคุณสามารถใช้numpy :

import numpy as np
a = np.array([-2, 1, 5, 3, 8, 5, 6])
b = [1, 2, 5]
print(list(a[b]))
# Result:
[1, 5, 5]

แต่จริงๆแล้วทางออกปัจจุบันของคุณดี มันอาจจะเป็นสิ่งที่ประณีตที่สุดของพวกเขาทั้งหมด

ทางเลือก:

>>> map(a.__getitem__, b)
[1, 5, 5]

>>> import operator
>>> operator.itemgetter(*b)(a)
(1, 5, 5)

วิธีแก้ปัญหาอื่นอาจทำได้ผ่านทางซีรีย์แพนด้า:

import pandas as pd

a = pd.Series([-2, 1, 5, 3, 8, 5, 6])
b = [1, 2, 5]
c = a[b]

จากนั้นคุณสามารถแปลง c กลับเป็นรายการหากคุณต้องการ:

c = list(c)

การทดสอบขั้นพื้นฐานและไม่ครอบคลุมมากเปรียบเทียบเวลาดำเนินการของคำตอบทั้งห้าที่ให้มา:

def numpyIndexValues(a, b):
    na = np.array(a)
    nb = np.array(b)
    out = list(na[nb])
    return out

def mapIndexValues(a, b):
    out = map(a.__getitem__, b)
    return list(out)

def getIndexValues(a, b):
    out = operator.itemgetter(*b)(a)
    return out

def pythonLoopOverlap(a, b):
    c = [ a[i] for i in b]
    return c

multipleListItemValues = lambda searchList, ind: [searchList[i] for i in ind]

ใช้อินพุตต่อไปนี้:

a = range(0, 10000000)
b = range(500, 500000)

simple python loop เป็นวิธีที่เร็วที่สุดในการดำเนินการแลมบ์ดาในไม่ช้าวินาที MapIndexValues ​​และ getIndexValues ​​ก็ค่อนข้างคล้ายกันกับวิธีการของ numpy ช้าลงอย่างมากหลังจากแปลงรายการเป็น numpy arrays วิธีการที่ numpyIndexValues ที่เร็วที่สุด

numpyIndexValues -> time:1.38940598 (when converted the lists to numpy arrays)
numpyIndexValues -> time:0.0193445 (using numpy array instead of python list as input, and conversion code removed)
mapIndexValues -> time:0.06477512099999999
getIndexValues -> time:0.06391049500000001
multipleListItemValues -> time:0.043773591
pythonLoopOverlap -> time:0.043021754999999995

ฉันแน่ใจว่าสิ่งนี้ได้รับการพิจารณาแล้ว: หากปริมาณของดัชนีใน b มีขนาดเล็กและคงที่เราสามารถเขียนผลลัพธ์ดังนี้:

c = [a[b[0]]] + [a[b[1]]] + [a[b[2]]]

หรือง่ายยิ่งขึ้นถ้าดัชนีของตัวเองเป็นค่าคงที่ ...

c = [a[1]] + [a[2]] + [a[5]]

หรือหากมีช่วงดัชนีที่ต่อเนื่องกัน ...

c = a[1:3] + [a[5]]

นี่เป็นวิธีที่ง่ายกว่า:

a = [-2,1,5,3,8,5,6]
b = [1,2,5]
c = [e for i, e in enumerate(a) if i in b]

คำตอบของฉันไม่ได้ใช้คอลเลกชัน numpy หรือหลาม

วิธีหนึ่งเล็กน้อยในการค้นหาองค์ประกอบจะเป็นดังนี้:

a = [-2, 1, 5, 3, 8, 5, 6]
b = [1, 2, 5]
c = [i for i in a if i in b]

ข้อเสียเปรียบ: วิธีนี้อาจใช้ไม่ได้กับรายการขนาดใหญ่ แนะนำให้ใช้ numpy สำหรับรายการขนาดใหญ่

ดัชนีคงที่และรายการเล็ก ๆ ?

อย่าลืมว่าถ้ารายการมีขนาดเล็กและดัชนีไม่เปลี่ยนแปลงเช่นในบางครั้งสิ่งที่ดีที่สุดคือการใช้ลำดับการเปิดออก :

_,a1,a2,_,_,a3,_ = a

ประสิทธิภาพดีขึ้นมากและคุณสามารถบันทึกโค้ดหนึ่งบรรทัด:

 %timeit _,a1,b1,_,_,c1,_ = a
10000000 loops, best of 3: 154 ns per loop 
%timeit itemgetter(*b)(a)
1000000 loops, best of 3: 753 ns per loop
 %timeit [ a[i] for i in b]
1000000 loops, best of 3: 777 ns per loop
 %timeit map(a.__getitem__, b)
1000000 loops, best of 3: 1.42 µs per loop

ชนิดของ pythonic way:

c = [x for x in a if a.index(x) in b]