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

小编典典

python

对于两个列表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)。

阅读 223

2020-12-20

共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]

2020-12-20