python 使用numpy實現從二維數組中找出行（子數組）為特定值的索引

1.問題描述

想從二維數組中找出某（幾）行滿足特定值的索引（Find matching rows in 2 dimensional numpy array）舉例如下：

　　有數組：a=np.array([[0, 0],[1, 0],[2, 0],[0, 1],[1, 1],[2, 1],[0, 2],[1, 2]])

　　想找到a中[0,1]所在的行索引，即3。

思考：在一維數組中找到某個值（val）的索引號，通過np.where(arr==val)即可滿足需求。而在二維數組中直接使用一維數組的方法卻不能滿足需求，搜索一番后發現stackoverflow上有一個一樣的問題，故記錄之。

參考：Find matching rows in 2 dimensional numpy array

2.解決辦法

a=np.array([[0, 0],[1, 0],[2, 0],[0, 1],[1, 1],[2, 1],[0, 2],[1, 2]])
rowIndex=np.where((a == (0, 1)).all(axis=1))
print('(0,1)在a中的索引為：',rowIndex)

輸出如下：