Python - “比较”将一个数组映射到另一个数组的简单方法

Question

我有一个数组a = [1, 2, 3, 4, 5, 6] 和b = [1, 3, 5]，我想映射a，这样对于a 中位于b 中的元素之间的每个元素，它都会映射到@ 的索引987654326@ 这是包含a 的上限。不是最好的文字解释，但这里有一个例子

a = 1 -> 0 because a <= first element of b
a = 2 -> 1 because b[0] < 2 <= b[1] and b[1] = 3
a = 3 -> 1 
a = 4 -> 2 because b[1] < 4 <= b[2]

所以我想要的最终产品是f(a, b) = [0, 1, 1, 2, 2, 2]

我知道我可以循环并解决它，但我想知道在 pandas/numpy 中是否有一种聪明、快速（矢量化）的方法来做到这一点

Answer 1

使用python的bisect模块：

from bisect import bisect_left

a = [1, 2, 3, 4, 5, 6]
b = [1, 3, 5]

def f(_a, _b):
  return [bisect_left(_b, i) for i in _a]

print(f(a, b))

bisect — 数组二等分算法

此模块支持按排序顺序维护列表，而无需在每次插入后对列表进行排序。对于具有昂贵比较操作的长列表项目，这可能是对更常见方法的改进。该模块被称为 bisect，因为它使用基本的二分算法来完成它的工作。源代码作为算法的工作示例可能最有用（边界条件已经正确！）。

提供以下功能：

bisect.bisect_left(a, x, lo=0, hi=len(a))

在 a 中定位 x 的插入点以保持排序顺序。参数 lo 和 hi 可用于指定应考虑的列表子集；默认情况下使用整个列表。如果 x 已经存在于 a 中，则插入点将位于任何现有条目之前（左侧）。 假设 a 已经排序，返回值适合用作list.insert() 的第一个参数。

返回的插入点 i 将数组 a 分成两半，all(val < x for val in a[lo:i]) 为左侧，all(val >= x for val in a[i:hi]) 为右侧。

参考： https://docs.python.org/3/library/bisect.html

Answer 2

二分法更快：解决方案假定列表已排序

a = [1, 2, 3, 4, 5, 6]
b = [1, 3, 5]

inds=[min(bisect_left(b,x),len(b)-1) for x in a]

返回

[0, 1, 1, 2, 2, 2]