Извлекать ненулевые элементы массива Python

numpy.nonzero(array)
#Return the indices of all the nonzero elements in the array
MitchAloha