在Python中查找矩阵内部向量的索引

Question

给出矩阵（Numpy数组）A和向量v。

A = np.array([[5,0],[1,-2],[0,2],[-1,3],[1,2]])
v = np.array([0,2])

获得向量v在矩阵A中的索引的最佳方法是什么（在这种情况下，应该获得2）。

Answer 1

这将返回A中的所有索引号：

print([i for i in range(len(A)) if (A[i]==v).all()])