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

Question

给定一个矩阵（ numpy数组） A和一个向量v 。

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

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

Answer 1

这样做：

np.argwhere((v == A).all(1))

Out[82]: array([[2]], dtype=int64)

Answer 2

如果最好的意思是最快的，经过彻底的实验，用户 hpaulj指出np.flatnonzero是np.flatnonzero的更快替代np.argwhere 。 你可以这样使用它：

np.flatnonzero((v==A).all(1))[0]

输出：

2