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]