python - python中的列表匹配：获取更大列表中子列表的索引

Question

对于两个列表，

a = [1, 2, 9, 3, 8, ...]   (no duplicate values in a, but a is very big)
b = [1, 9, 1,...]          (set(b) is a subset of set(a), 1<<len(b)<<len(a)) 

indices = get_indices_of_a(a, b)

如何让get_indices_of_a返回？有没有比 using 更快的方法，这需要太长时间？indices = [0, 2, 0,...]array(a)[indices] = ba.index

创建b一个集合是匹配列表和返回索引的快速方法（请参阅比较python中的两个列表并返回匹配值的索引），但在这种情况下它将丢失第二个的索引1以及索引的序列。

Answer 1

一种快速方法（何时a是一个大列表）将使用 dict 将值映射a到索引：

>>> index_dict = dict((value, idx) for idx,value in enumerate(a))
>>> [index_dict[x] for x in b]
[0, 2, 0]

a.index与使用需要二次时间相比，这在平均情况下将花费线性时间。

Answer 2

假设我们正在使用较小的列表，这很简单：

>>> a = [1, 2, 9, 3, 8] 
>>> b = [1, 9, 1] 
>>> [a.index(item) for item in b]
[0, 2, 0]

在较大的列表中，这将变得非常昂贵。

（如果有重复，第一次出现将始终是结果列表中引用的那个，如果not set(b) <= set(a)，你会得到一个 ValueError）。