查找给定的1D np.array在2D np.array中的位置

Question

我需要找到一条线在矩阵中的位置。

假设我有2D numpy数组 A 和1D numpy数组 B 。我想在 A 中检索等于 B 的行的索引。

A=np.array([[0,1,2],[3,4,5],[15,7,8]])
B=np.array([3,4,5])

理想情况下，结果应为 1 ，如

A[1]==B

有没有一种方法可以通过numpy方法实现？我正在处理大型矩阵，因此我希望避免for循环。

谢谢