np.argwhere( a )
Find the indices of array elements that are non-zero, grouped by element.
返回非0的数组元组的索引,其中a是要索引数组的条件。
eg:输出数组中所有大于1的数字的索引值。
x=np.arange(6).reshape(2,3)
x
array([[0,1,2],
[3,4,5]])
np.argwhere(x>1)
array([[0,2],
[1,0]
[1,1]
[1,2]],dtype=int64)