numpy数组的逐元素逻辑比较

Question

我有两个形状相同的numpy数组。这两个数组里的元素是从[0,N]中随机生成的整数。我需要检查这两个数组中同一位置的元素是否相等。

我想要的输出是那些相同元素的位置。

示例代码：

A=np.array([0,1])
B=np.array([1,0])
C=np.array([1,1])
np.any_elemenwise(A,B)
np.any_elemenwise(A,C)
np.any_elemenwise(A,A)

期望的输出：

[]
[1]
[0,1]

我可以写一个循环，逐个检查所有元素，但我觉得可以用更快的方法来得到想要的结果。

Answer 1

编辑：问题已经改变。

你只需要计算 np.where(v1==v2)[0] 的结果。