numpy.nonzero()
numpy.nonzero
-
numpy.nonzero(a)[source] -
Return the indices of the elements that are non-zero.
Returns a tuple of arrays, one for each dimension of
a, containing the indices of the non-zero elements in that dimension. The values inaare always tested and returned in row-major, C-style order. The corresponding non-zero values can be obtained with:a[nonzero(a)]
To group the indices by element, rather than dimension, use:
transpose(nonzero(a))
The result of this is always a 2-D array, with a row for each non-zero element.
登录查看完整内容
