색인을 알고있는 목록의 여러 요소에 액세스하십시오.

Question

주어진 목록에서 색인을 아는 일부 요소를 선택해야합니다. 주어진리스트 [-2, 1, 5, 3, 8, 5, 6]에서 인덱스 1, 2, 5가있는 요소를 포함하는 새 목록을 만들고 싶습니다. 내가 한 것은 :

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

그것을 할 수있는 더 좋은 방법이 있나요? c = a [b]와 같은 무엇인가?

Answer 1

당신은 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] 

---

하지만 실제로는 현재 해결책이 좋습니다. 그것은 아마도 그들 모두에서 가장 깨끗합니다.

Answer 2

대안 :

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

---

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

Answer 3

내 대답은 numpy 또는 python 모음을 사용하지 않습니다. 다음과 같은 요소가 될 것이다 찾을 수

한 사소한 방법 :

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

단점 :이 방법은 큰 목록에 대한 작동하지 않을 수 있습니다. 큰 목록에는 numpy를 사용하는 것이 좋습니다. 다음 입력을 사용

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] 

: 다섯 개 공급 응답의 실행 시간을 비교

Answer 4

기본 아니라 매우 광범위한 테스트

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

간단한 파이썬 루프이었다 람다 동작 A의 빠른 두 번째 닫기, mapIndexValues ​​및 getIndexValues는 numpy 메소드와 일관되게 매우 유사했습니다. 목록이 numpy 배열로 변환 된 후 상당히 느립니다. 데이터가 이미 numpy 배열에있는 경우 numpy가있는 numpyIndexValues ​​메소드 .array 변환이 가장 빠릅니다.

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 

Answer 5

나는이 이미 고려 된 확신 :

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

또는 인덱스 경우 더 간단 : B의 ​​인덱스의 양이 작고 일정한 경우, 하나는 같은 결과를 쓸 수 그 자체가 ...

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

상수 또는 인덱스의 연속 범위가있는 경우 ...

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

Answer 6

또 다른 해결책은 팬더 시리즈를 통해 수 :

이

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)