比较python中的两个列表并返回匹配值的索引

Question

对于两个列表a和b，如何获取两个列表中出现的值的索引？例如，

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

return_indices_of_a(a, b) 

将返回[0,4]，与(a[0],a[4]) = (1,5)。

Answer 1

要做到这一点，最好的方法是制作b a 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的。