numpy - 将数组与 Numpy 中的行进行匹配

Question

我有一个形状为(50,3)的数组“A”和另一个形状为(1,3)的数组“B”。

其实这个B是A中的一行，所以我需要找到它的行位置。

我使用了np.where(A==B)，但它给出了按元素搜索的位置。例如，下面是我得到的结果:

>>> np.where(A == B)
(array([ 3,  3,  3, 30, 37, 44]), array([0, 1, 2, 1, 2, 0]))

实际上 B 是 A 中的第四行(在我的例子中)。但上面的结果给出了 (3,0)(3,1)(3,2) 等，它们是按元素匹配的。

取而代之的是，我需要一个答案“3”，这是 B 在 A 中整体搜索时获得的答案，并且它还删除了其他像 (30,1)(37,2)... 的部分匹配.

如何在 Numpy 中执行此操作？

谢谢。

Answer 1

您可以指定轴:

numpy.where((A == B).all(axis=1))