В Python, как мне индексировать список другим списком?

Question

Я хотел бы проиндексировать список с другим списком, как этот

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

Answer 1

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

Answer 2

Если вы используете 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')

... и, вероятно, намного быстрее (если производительности достаточно, чтобы беспокоиться о крошечном импорте)

Answer 3

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

Answer 4

Функциональный подход:

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

from operator import itemgetter

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

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

Answer 6

Вы также можете использовать __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']

Answer 7

Моя проблема: найти индексы списка.

      L = makelist() # Returns a list of different objects
La = np.array(L, dtype = object) # add dtype!
for c in chunks:
    L_ = La[c] # Since La is array, this works.

Answer 8

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

Я бы использовал Dict add желательно keys в index