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