In Python, come posso indicizzare un elenco con un altro elenco?

130

Vorrei indicizzare un elenco con un altro elenco come questo

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

e T dovrebbe finire per essere un elenco contenente ['a', 'd', 'h'].

C'è un modo migliore di

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

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

python list indexing

— Daniel Andrén
fonte

241

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

— furgone
fonte

7

È più veloce di un for-loop o solo più corto?

— Daniel Andrén,

10

@daniel: entrambi + consigliati

— SilentGhost,

14

Un test di cronometraggio rapido (senza pysco o altro, quindi fai quello che vuoi) ha mostrato la comprensione dell'elenco 2,5 volte più veloce del loop (1000 elementi, ripetuto 10000 volte).

— James Hopkin,

2

(l'uso di map e lambda è ancora più lento - prevedibile, poiché chiama una funzione per ogni iterazione)

— James Hopkin,

+1 Se l'elenco di indicizzazione è arbitrario, allora la comrpehlist di elenco è la strada. Penso che, quando possibile, il che sembra non essere il caso qui, le sezioni sono ancora più veloci.

— Jaime,

41

Se si utilizza Numpy, è possibile eseguire il taglio esteso in questo modo:

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

... ed è probabilmente molto più veloce (se le prestazioni sono abbastanza importanti da disturbare con l'importazione numpy)

— Paolo
fonte

5

Il mio rapido test di timeit ha dimostrato che l'utilizzo di np.array è in realtà quasi 3 volte più lento (inclusa la conversione in array).

— Andrzej Pronobis,

Funziona meglio se è necessario convertirlo per operazioni di array comunque. Troppo tempo per le normali operazioni dell'elenco.

— frankliuao,

9

Un approccio funzionale:

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

from operator import itemgetter

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

— Padraic Cunningham
fonte

Questo non funzionerà se bcapita di contenere solo un oggetto.

— blhsing

7

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

— Mehrdad Afshari
fonte

6

doveva essere convertito in elenco in py3k

— SilentGhost il

5

Non ero contento di nessuno di questi approcci, quindi ho creato una Flexlistclasse che consente l'indicizzazione flessibile, sia per numero intero, slice o indice-elenco:

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

Che, per il tuo esempio, utilizzeresti come:

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

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

— jedwards
fonte

che dimostra anche la potenza e la flessibilità di Python!

— crowie,

È così facile estenderlo anche per il codice esistente. Basta chiamare existing_list = Flexlist(existing_list)e abbiamo la funzionalità richiesta senza rompere alcun codice

— Yesh

1

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

Vorrei utilizzare un Dict adddesiderato keysperindex

— user4749532
fonte

0

È inoltre possibile utilizzare il __getitem__metodo combinato con mapil seguente:

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

— David S
fonte