Pythonで2つのリストを比較し、一致した値のインデックスを返す

Question

2つのリストaとbに対して、両方に表示される値のインデックスを取得するにはどうすればよいですか？例えば、

a = [1, 2, 3, 4, 5] 
b = [9, 7, 6, 5, 1, 0] 

return_indices_of_a(a, b) 

は(a[0],a[4]) = (1,5)で、[0,4]を返します。

Answer 1

bをsetにするのが最も良い方法は、メンバーシップの確認だけであるためです。大きなリストでは

>>> a = [1, 2, 3, 4, 5] 
>>> b = set([9, 7, 6, 5, 1, 0]) 
>>> [i for i, item in enumerate(a) if item in b] 
[0, 4] 

Answer 2

def return_indices_of_a(a, b): 
    b_set = set(b) 
    return [i for i, v in enumerate(a) if v in b_set] 

Answer 3

これは助けになることがあります。

for item in a: 
index.append(bisect.bisect(b,item)) 
    idx = np.unique(index).tolist() 

はnumpyのをインポートするようにしてください。