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]