检查两个numpy数组python中有多少元素相等

check how many elements are equal in two numpy arrays python

提问by Shai Zarzewski

I have two numpy arrays with number (Same length), and I want to count how many elements are equal between those two array (equal = same value and position in array)

我有两个带有数字（相同长度）的 numpy 数组，我想计算这两个数组之间有多少个元素相等（相等 = 数组中的相同值和位置）

A = [1, 2, 3, 4]
B = [1, 2, 4, 3]

then I want the return value to be 2 (just 1&2 are equal in position and value)

然后我希望返回值为 2（只有 1&2 在位置和值上相等）

采纳答案by falsetru

Using numpy.sum:

使用numpy.sum：

>>> import numpy as np
>>> a = np.array([1, 2, 3, 4])
>>> b = np.array([1, 2, 4, 3])
>>> np.sum(a == b)
2
>>> (a == b).sum()
2

回答by jdehesa

As long as both arrays are guaranteed to have the same length, you can do it with:

只要保证两个数组的长度相同，您就可以这样做：

np.count_nonzero(A==B)