是否存在类似np.isin且支持容差的高效数组元素匹配方法？

方法1：NumPy向量化广播操作（适合中等规模数组） #

利用NumPy的广播特性实现无显式循环的高效计算，避免逐元素遍历的低效问题：

import numpy as np

array1 = [1, 2, 3, 4, 5, 6]
array2 = [2, 8, 1.00001, 1.1]
tolerance = 0.001

arr1 = np.array(array1)
arr2 = np.array(array2)

# 生成所有元素对的绝对差矩阵（形状为(len(arr1), len(arr2))）
diff_matrix = np.abs(arr1[:, np.newaxis] - arr2)
# 检查每个array1元素是否存在匹配的array2元素
has_match = np.any(diff_matrix <= tolerance, axis=1)
# 提取匹配的索引
places = np.where(has_match)[0].tolist()

print(places)  # 输出: [0, 1]

核心逻辑是通过广播将一维数组扩展为二维差矩阵，再用np.any快速判断每行是否存在符合容差的元素，最终用np.where筛选出目标索引。

方法2：排序+二分查找（适合超大数组） #

当数组元素量级达到数万甚至更多时，排序后用二分查找能将时间复杂度从O(n*m)降至O(n log n + m log n)，大幅提升效率：

import numpy as np

array1 = [1, 2, 3, 4, 5, 6]
array2 = [2, 8, 1.00001, 1.1]
tolerance = 0.001

arr1 = np.array(array1)
arr2_sorted = np.sort(array2)

places = []
for idx, num in enumerate(arr1):
    # 找到num在排序后array2中的插入位置
    pos = np.searchsorted(arr2_sorted, num)
    # 检查前后相邻元素（处理边界情况）
    candidates = []
    if pos > 0:
        candidates.append(arr2_sorted[pos-1])
    if pos < len(arr2_sorted):
        candidates.append(arr2_sorted[pos])
    # 判断是否有符合容差的候选元素
    for cand in candidates:
        if abs(num - cand) <= tolerance:
            places.append(idx)
            break

print(places)  # 输出: [0, 1]

通过np.searchsorted快速定位最接近的元素位置，仅需检查该位置前后的元素即可完成匹配判断，避免了全量元素对比。

方法3：纯Python实现（无需NumPy依赖） #

如果不想依赖NumPy，可使用Python标准库的bisect模块实现二分查找逻辑：

import bisect

array1 = [1, 2, 3, 4, 5, 6]
array2 = [2, 8, 1.00001, 1.1]
tolerance = 0.001

arr2_sorted = sorted(array2)
places = []

for idx, num in enumerate(array1):
    pos = bisect.bisect_left(arr2_sorted, num)
    match_found = False
    # 检查左侧相邻元素
    if pos > 0 and abs(num - arr2_sorted[pos-1]) <= tolerance:
        match_found = True
    # 左侧无匹配则检查右侧元素
    if not match_found and pos < len(arr2_sorted) and abs(num - arr2_sorted[pos]) <= tolerance:
        match_found = True
    if match_found:
        places.append(idx)

print(places)  # 输出: [0, 1]

该方法完全基于Python标准库实现，逻辑简洁，同时保持了远高于全量循环的效率。

内容的提问来源于stack exchange，提问作者celery