numpy.flatnonzero() en Python

numpy.flatnonzero()La función se utiliza para calcular índices que no son cero en la versión aplanada de arr.

Sintaxis: numpy.flatnonzero(arr)

Parámetros:
arr: [array_like] Array de entrada.

Retorno: array de salida ndarray
, que contiene los índices de los elementos de arr.ravel() que no son cero.

Código #1: Trabajando

# Python program explaining
# flatnonzero() function
  
import numpy as geek
arr = geek.arange(-3, 4)
  
print ("Input  array : ", arr)
    
out_arr = geek.flatnonzero(arr)
print ("Indices of non zero elements : ", out_arr) 

Producción :

Input  array :  [-3 -2 -1  0  1  2  3]
Indices of non zero elements :  [0 1 2 4 5 6]

 
Código # 2: Usar los índices de los elementos distintos de cero como una array de índice.

# Python program using the indices of the non-zero 
# elements as an index array to extract these elements
  
out_arr = arr.ravel()[geek.flatnonzero(arr)]
  
print ("Output array of non-zero number: ", out_arr) 

Producción :

Output array of non-zero number:  [-3 -2 -1  1  2  3]