IT

Python에서 다른 목록으로해야 할 일이 있습니까?

lottoking 2020. 8. 14. 08:02
반응형

Python에서 다른 목록으로해야 할 일이 있습니까?


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

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']

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

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 가져 오기 오기를 귀찮게 할 충분한 경우)


기능적 접근 :

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

from operator import itemgetter

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


T = map(lambda i: L[i], Idx)

나는 접근 방식 방식에 만족하지 갑자기 기 때문에 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']

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

나는 Dict add원하는 keys것을 사용합니다index

참고 URL : https://stackoverflow.com/questions/1012185/in-python-how-do-i-index-a-list-with-another-list

반응형