np.argwhere（）的用法

np.argwhere( a ) 

Find the indices of array elements that are non-zero, grouped by element.

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

返回数组中所有大于1的数字的索引值。