在给定相同索引和返回匹配数的情况下，检查两个数组中的匹配元素

Question

给定两个数组，是否有一种numpy非循环方法来检查数组之间的每个第i个索引是否匹配，即如果a[i]==b[i]，是否检查每个i？

a = np.array([1,2,3,4,5,6,7,8])
b = np.array([2,3,4,5,6,7,8,9])
Output:  0 matches

我想这已经被问过了，但是我找不到我要找的东西，如果是的话，我很抱歉。

Answer 1

其他答案的另一个变体：

np.flatnonzero(a == b)

Answer 2

您可以使用numpy.where提取两个值满足指定条件的索引：

import numpy

a = numpy.array([0, 1, 2, 3, 4, 5, 6])
b = numpy.array([6, 5, 4, 3, 2, 1, 6])
numpy.where(a==b)

(array([3, 6]),)

Answer 3

试试这个：

np.arange(len(a))[a==b]

它创建一个从0到长度a的新数组，表示索引。然后使用a==b对数组进行切片，返回a和b相同的索引。

此外，来自@Reblochon-Masque：

您可以使用numpy.where提取两个值满足指定条件的索引：

import numpy

a = numpy.array([0, 1, 2, 3, 4, 5, 6])
b = numpy.array([6, 5, 4, 3, 2, 1, 6])
numpy.where(a==b)

(array([3, 6]),)