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

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

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

那么我希望返回值为2(只有1和2在位置和值上相等)

1> falsetru..：

---

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

---

@NicolasSchejtman答案中的解决方案应该适用于2D阵列.试试`a = np.array([[1,2,0],[3,4,0]]); b = np.array([[1,9,0],[9,4,0]]); print((a == b).sum())`

2> jdehesa..：

---

只要两个阵列保证长度相同,您就可以使用:

np.count_nonzero(A==B)

---

@Euler_Salter假设你想要计算具有相同值和位置的元素,我猜想就像`s = min(len(A),len(B)); count = np.count_nonzero(A [:s] == B [:s])`.