比较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一个,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]