[Python] Numpy.nonzero

NumPy. Nonzero (a)

Return the indices of the elements that is non-zero.

Returns a tuple of arrays, one for each dimension of a, containing the indices of the Non-zero elements in that D Imension. The values in ais always tested and returned in Row-major, C-style order. The corresponding Non-zero values can be obtained with:

Import NumPy as NPA = Np.eye (3);p rint Np.nonzero (a) # (array ([0, 1, 2]), array ([0, 1, 2])

　　

Print Np.transpose (Np.nonzero (a)) #[[0 0]# [1 1]# [2 2]]

　　

[Python] Numpy.nonzero