Numpy 1.11
Numpy 1.11
  • 请登录查看
字典 我要纠错

MaskedArray.nonzero()

numpy.ma.MaskedArray.nonzero

MaskedArray.nonzero() [source]

Return the indices of unmasked elements that are not zero.

Returns a tuple of arrays, one for each dimension, containing the indices of the non-zero elements in that dimension. The corresponding non-zero values can be obtained with:

a[a.nonzero()]

To group the indices by element, rather than dimension, use instead:

np.transpose(a.nonzero())

The result of this is always a 2d array, with a row for each non-zero element.

登录查看完整内容