python 两个列表相互映射_Python：两个列表之间的快速映射和查找

我认为建议您使用哈希表来代替

numpy.in1d,它使用O(n log n)合并排序作为预处理步骤.

>>> A = [500, 300, 400, 200, 100]

>>> index = { k:i for i,k in enumerate(A, 1) }

>>> random_list = [200, 100, 50]

>>> [i for i,x in enumerate(random_list) if x in index]

[0, 1]

>>> map(index.get, random_list)

[4, 5, None]

>>> filter(None, map(index.get, random_list))

[4, 5]

这是Python 2,在Python 3中映射并过滤返回类似于生成器的对象,因此如果要将结果作为列表获取,则需要围绕过滤器的列表.

我试图尽可能地使用内置函数来将计算负担推到C端(假设你使用CPython).所有的名字都是预先解决的,所以它应该非常快.

通常,为了获得最佳性能,您可能需要考虑使用PyPy,这是一个使用JIT编译的很好的替代Python实现.

比较多种方法的基准从来都不是一个坏主意：

import sys

is_pypy = '__pypy__' in sys.builtin_module_names

import timeit

import random

if not is_pypy:

import numpy

import bisect

n = 10000