Dans Python, comment indexer une liste avec une autre liste ?

Question

J'aimerais indexer une liste avec une autre liste comme celle-ci

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

et T devrait finir par être une liste contenant ['a', 'd', 'h'].

Y a-t-il un meilleur moyen que

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

Si vous utilisez numpy, vous pouvez effectuer un découpage étendu comme ceci :

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

...et est probablement beaucoup plus rapide (si la performance est assez d'un souci à se préoccuper de l'importation engourdie)

Answer 3

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

Answer 4

Une approche fonctionnelle :

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

Je n'étais satisfaite d'aucune de ces approches, alors j'ai trouvé une classe Flexlist qui permet une indexation flexible, soit par entier, tranche ou index-list :

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

Que, pour votre exemple, vous utiliseriez comme :

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

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