我有两个相同形状的 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]
我可以编写一个循环遍历所有元素,但我认为可以更快地实现所需的输出。
最佳答案
编辑:问题变了。
你只想评估np.where(v1==v2)[0]
关于python - numpy 数组的逐元素逻辑比较,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/22605554/