lista partita in python: ottenere indici di un sotto-elenco in una lista più ampia

Question

Per due liste,

a = [1, 2, 9, 3, 8, ...] (no duplicate values in a, but a is very big) 
b = [1, 9, 1,...]   (set(b) is a subset of set(a), 1<<len(b)<<len(a)) 

indices = get_indices_of_a(a, b) 

come lasciare get_indices_of_a ritorno indices = [0, 2, 0,...] con array(a)[indices] = b? C'è un metodo più veloce rispetto all'utilizzo di a.index, che richiede troppo tempo?

Fare b un insieme è un metodo rapido di corrispondenza elenchi e ritorno indici (vedi compare two lists in python and return indices of matched values), ma perderà l'indice del secondo 1 così come la sequenza degli indici in questo caso.

Answer 1

Procedimento veloce (quando a è una grande lista) sarebbe utilizzando un dict per mappare i valori in a agli indici:

>>> index_dict = dict((value, idx) for idx,value in enumerate(a)) 
>>> [index_dict[x] for x in b] 
[0, 2, 0] 

Questo richiederà tempo lineare nel caso medio, rispetto all'utilizzo a.index cui prenderebbe il tempo quadratico

Answer 2

Presumendo che stiamo lavorando con le liste più piccole, questo è facile come:

>>> a = [1, 2, 9, 3, 8] 
>>> b = [1, 9, 1] 
>>> [a.index(item) for item in b] 
[0, 2, 0] 

sulle liste più grandi, questo diventerà piuttosto costoso.

(Se ci sono duplicati, la prima occorrenza sarà sempre quella a cui si fa riferimento nell'elenco risultante, se not set(b) <= set(a), si otterrà un errore ValueError).