파이썬에서 다른 목록으로 목록을 어떻게 색인화합니까?


130

이 같은 다른 목록으로 목록을 색인하고 싶습니다.

L = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
Idx = [0, 3, 7]
T = L[ Idx ]

그리고 T는 [ 'a', 'd', 'h']를 포함하는 목록이되어야합니다.

보다 좋은 방법이 있습니까

T = []
for i in Idx:
    T.append(L[i])

print T
# Gives result ['a', 'd', 'h']

답변:


241
T = [L[i] for i in Idx]

7
for-loop보다 빠르거나 짧습니까?
Daniel Andrén

10
@daniel : 둘 다 + 권장
SilentGhost 2016 년

14
빠른 타이밍 테스트 (pysco 또는 다른 것 없음)를 사용하면 루프보다 1000 배 빠른 목록 이해도 (1000 개 요소, 10000 회 반복)를 알 수 있습니다.
James Hopkin 2016 년

2
(지도를 사용하고 람다도 느립니다 - 그것은 각 반복에 대한 함수를 호출하기 때문에, 예상 할 수있는)
제임스 Hopkin

+1 인덱싱 목록이 임의적이면 목록 구성이 좋습니다. 나는 가능하다면 여기에서는 그렇지 않은 것처럼 보이지만 슬라이스가 훨씬 빠릅니다.
Jaime

41

numpy를 사용하는 경우 다음과 같이 확장 슬라이싱을 수행 할 수 있습니다.

>>> import numpy
>>> a=numpy.array(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
>>> Idx = [0, 3, 7]
>>> a[Idx]
array(['a', 'd', 'h'], 
      dtype='|S1')

... 그리고 아마도 더 빠를 것입니다 (numpy 가져 오기를 방해하기에 성능이 충분할 경우)


5
나의 빠른 timeit 테스트는 np.array를 사용하는 것이 실제로 거의 3 배 느리다는 것을 보여주었습니다 (배열로의 변환 포함).
Andrzej Pronobis

어쨌든 배열 작업을 위해 변환 해야하는 경우 더 잘 작동합니다. 일반 목록 작업에 시간이 너무 많이 걸립니다.
frankliuao

9

기능적인 접근 방식 :

a = [1,"A", 34, -123, "Hello", 12]
b = [0, 2, 5]

from operator import itemgetter

print(list(itemgetter(*b)(a)))
[1, 34, 12]

b하나의 항목 만 포함 하면 작동하지 않습니다 .
blhsing


5

나는이 접근법 중 어느 것에도 만족하지 않았으므로 Flexlist정수, 슬라이스 또는 색인 목록으로 유연한 색인을 생성 할 수 있는 클래스를 생각해 냈습니다 .

class Flexlist(list):
    def __getitem__(self, keys):
        if isinstance(keys, (int, slice)): return list.__getitem__(self, keys)
        return [self[k] for k in keys]

예를 들어 다음과 같이 사용하십시오.

L = Flexlist(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
Idx = [0, 3, 7]
T = L[ Idx ]

print(T)  # ['a', 'd', 'h']

또한 파이썬의 힘과 유연성을 보여줍니다!
crowie

기존 코드에 대해서도이를 쉽게 확장 할 수 있습니다. 전화 만하면 existing_list = Flexlist(existing_list)코드를 깨지 않고도 필요한 기능을 사용할 수 있습니다

1
L= {'a':'a','d':'d', 'h':'h'}
index= ['a','d','h'] 
for keys in index:
    print(L[keys])

나는 Dict add원하는 keys것을 사용할 것이다index


0

다음 __getitem__map같은 방법을 결합하여 사용할 수도 있습니다 .

L = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
Idx = [0, 3, 7]
res = list(map(L.__getitem__, Idx))
print(res)
# ['a', 'd', 'h']
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.