No Python, como indexar uma lista com outra lista?

130

Gostaria de indexar uma lista com outra lista como esta

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

e T deve acabar sendo uma lista contendo ['a', 'd', 'h'].

Existe uma maneira melhor do que

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]

— furgão
fonte

6

Isso é mais rápido que um loop for ou apenas mais curto?

— 18119 Daniel Andrén

9

@daniel: both + recommended

— SilentGhost 18/06/09

13

Um teste de tempo rápido (sem pysco ou qualquer coisa, faça o que você quiser) mostrou a compreensão da lista 2,5 vezes mais rápida que o loop (1000 elementos, repetidos 10000 vezes).

— James Hopkin

2

(usando mapa e um lambda é ainda mais lento - de se esperar, uma vez que chama uma função para cada iteração)

— James Hopkin

+1 Se a lista de indexação for arbitrária, então a extensão da lista é o caminho. Penso que, quando possível, o que parece não ser o caso aqui, as fatias são ainda mais rápidas.

— Jaime

40

Se você estiver usando numpy, poderá executar fatias estendidas assim:

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

... e provavelmente é muito mais rápido (se o desempenho for uma preocupação suficiente para se preocupar com a importação numpy)

— Paulo
fonte

5

Meu teste rápido de timeit mostrou que o uso do np.array é quase três vezes mais lento (incluindo a conversão em array).

— Andrzej Pronobis

Funciona melhor se você precisar convertê-lo para operações de matriz de qualquer maneira. Demorado demais para operações regulares da lista.

— frankliuao 3/01/19

9

Uma abordagem 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
fonte

Isso não funcionará se bcontiver apenas um item.

— blhsing 19/12/19

7

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

— Mehrdad Afshari
fonte

6

precisava ser convertida a lista em Py3k

— SilentGhost

5

Como eu não estava satisfeito com nenhuma dessas abordagens, criei uma Flexlistclasse que permite a indexação flexível, seja por número inteiro, fatia ou lista de índices:

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 exemplo, você usaria como:

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

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

— jedwards
fonte

o que também demonstra o poder e a flexibilidade do Python!

— 27618 crowie

É muito fácil estender isso também para o código existente. Basta ligar existing_list = Flexlist(existing_list)e nós temos a funcionalidade necessária sem quebrar nenhum código

— Yesh 9/01

1

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

Eu usaria um Dict adddesejado keysparaindex

— user4749532
fonte

0

Você também pode usar o __getitem__método combinado com mapo seguinte:

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