numpy数组的元素逻辑比较答案

【问题标题】：Elementwise logical comparison of numpy arraysnumpy数组的元素逻辑比较
【发布时间】：2014-03-24 09:35:06
【问题描述】：

我有两个相同形状的 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]

我可以编写一个循环遍历所有元素，但我认为可以更快地实现所需的输出。

【问题讨论】：

【解决方案1】：

编辑：问题变了。

你只想评估np.where(v1==v2)[0]

【讨论】：