在 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