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

Question

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

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

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

Answer 1

使用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

Answer 2

只要两个数组都保证长度相同，就可以使用：

np.count_nonzero(A==B)