python学习：np.argwhere()用法

python np.argwhere()用法

np.argwhere( a )

返回非0的数组元组的索引，其中a是要索引数组的条件。

eg:

A=np.array([0,1,1,0,0,1,1,0,0])
np.argwhere(A)
#输出为：
array([[1],
       [2],
       [5],
       [6]])
       
np.argwhere(A)[:,0] 
array([1, 2, 5, 6])

说明：np.argwhere输出是一列元素，使用[:,0]变成一行元素