np.argwhere( a )

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

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