是否有类似于 np.isin 的 numpy 函数，它允许容差而不是要求值相同？

Question

我有两个不同大小的数组，想要确定一个数组的元素可以在另一个数组中找到。我希望能够允许元素之间存在容差。目标是这样的：

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

places = [mystery function](array1, array2, tolerance = 0.001)

在

places = [0,1]

中返回索引

array1

。

我能得到的最接近的是

np.isin

，它允许不同大小和顺序的数组，但不允许容差。 （也尝试过

np.allclose

，但形状和顺序不匹配是一个问题）。当然，这可以通过循环来完成，但我的实际数组有数千个元素长，因此循环不实用。

（它也不必是 numpy 函数——实际上只是比循环更有效的任何东西）

预先感谢您的帮助！