Trong Python, làm cách nào để lập chỉ mục một danh sách với một danh sách khác?

Tôi muốn lập chỉ mục một danh sách với một danh sách khác như thế này

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

và T cuối cùng sẽ là một danh sách chứa ['a', 'd', 'h'].

Có cách nào tốt hơn

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

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

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

Nếu bạn đang sử dụng numpy, bạn có thể thực hiện cắt lát mở rộng như thế:

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

... và có lẽ nhanh hơn nhiều (nếu hiệu suất đủ để lo lắng với việc nhập khẩu khó chịu)

Một cách tiếp cận chức năng:

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)

Tôi không hài lòng với bất kỳ cách tiếp cận nào trong số này, vì vậy tôi đã đưa ra một Flexlistlớp cho phép lập chỉ mục linh hoạt, theo số nguyên, lát hoặc danh sách chỉ mục:

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

Mà, ví dụ của bạn, bạn sẽ sử dụng như:

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

Tôi sẽ sử dụng một Dict addmong muốn keysđểindex

Bạn cũng có thể sử dụng __getitem__phương thức kết hợp mapnhư sau:

L = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
Idx = [0, 3, 7]
res = list(map(L.__getitem__, Idx))
print(res)
# ['a', 'd', 'h']