이 같은 다른 목록으로 목록을 색인하고 싶습니다.
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
답변
다음 __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']