Numpy 检查两个 arrays 的元素大致相等的位置

Question

I have two numpy arrays with floating point values and I am trying to find the indices where the numbers are approximately equal (floating point comparisons).

So something like:

x = np.random.rand(3)
y = np.random.rand(3)

x[2] = y[2]

# Do the comparison and it should return 2 as the index

I tried something like

np.where(np.allclose(x, y))

However, this returns an empty array. If I do:

np.where(x == y)  # This is fine.

I tried using a combination of numpy.where and numpy.allclose but could not make it work. Of course, I can do it with a loop but that seems tedious and unpythonic.

Answer 1

你要找的是np.isclose ：

np.where(np.isclose(x, y))

Answer 2

你可以随时使用依赖的东西：

np.where( np.abs(x-y) < epsilon )