En Python, ¿cómo indexo una lista con otra lista?

130

Me gustaría indexar una lista con otra lista como esta

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

y T debería terminar siendo una lista que contiene ['a', 'd', 'h'].

¿Hay una mejor manera que

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

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

python list indexing

— Daniel Andrén
fuente

241

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

— camioneta
fuente

77

¿Es esto más rápido que un bucle for o solo más corto?

— Daniel Andrén

10

@daniel: ambos + recomendado

— SilentGhost

14

Una prueba de sincronización rápida (sin pysco ni nada, así que haga lo que quiera) mostró la comprensión de la lista 2.5 veces más rápido que el ciclo (1000 elementos, repetidos 10000 veces).

— James Hopkin

2

(usando un mapa y lambda es aún más lento - es de esperar, ya que llama a una función para cada iteración)

— James Hopkin

+1 Si la lista de indexación es arbitraria, entonces la comprensión de la lista es el camino. Sin embargo, creo que, cuando sea posible, lo que parece no ser el caso aquí, los cortes son aún más rápidos.

— Jaime

41

Si está usando numpy, puede realizar un corte extendido así:

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

... y probablemente sea mucho más rápido (si el rendimiento es una preocupación suficiente para molestarse con la importación numpy)

— Pablo
fuente

55

Mi prueba rápida de timeit mostró que usar np.array es en realidad casi 3 veces más lento (incluida la conversión a matriz).

— Andrzej Pronobis

Funciona mejor si necesita convertirlo para operaciones de matriz de todos modos. Demasiado tiempo para las operaciones regulares de la lista.

— frankliuao

9

Un enfoque funcional:

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
fuente

Esto no funcionará si bcontiene solo un elemento.

— blhsing

7

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

— Mehrdad Afshari
fuente

66

necesaria para ser convertidos a la lista en py3k

— SilentGhost

5

No estaba contento con ninguno de estos enfoques, por lo que se me ocurrió una Flexlistclase que permite una indexación flexible, ya sea por entero, por segmento o por índice:

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, por ejemplo, usarías como:

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

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

— jedwards
fuente

que también demuestra el poder y la flexibilidad de Python!

— crowie

Es muy fácil extender esto también para el código existente. Simplemente llame existing_list = Flexlist(existing_list)y tenemos la funcionalidad requerida sin descifrar ningún código

— Yesh

1

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

Yo usaría un Dict adddeseado keysparaindex

— usuario4749532
fuente

0

También puede usar el __getitem__método combinado con maplo siguiente:

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
fuente