python 元组列表中的元素索引

Question

问题

给定 2 个列表 A 和 B，我想获取列表 B 中存在的列表 A 中所有元素的索引。每个元素都是一个元组。

我正在使用大小为 40,000 个元素左右的列表。

样品案例

输入：

A = [(1,2),(3,4),(5,6),(7,8)]

B = [(1,2),(3,4),(5,6)]

预期 output：

[0,1,2]

尝试的解决方案

我尝试了两种解决方案：

1) 使用 map function

m = map(a.index,b)
list(m)

2) 使用列表理解

m = [a.index(item) for item in b if item in a]

这些方法似乎花费了太多时间。 有没有其他方法可以做到这一点？

Answer 1

以下将是您最好的选择。 我正在使用一个集合（即set(B) ），因为可以在 O(1) 时间复杂度中完成对特定元组的搜索。

m = [index for index, tuple in enumerate(A) if tuple in set(B)]