numpy.argwhere() La función se utiliza para encontrar los índices de los elementos de la array que no son cero, agrupados por elemento.
Sintaxis: numpy.argwhere(arr)
Parámetros:
arr: [array_like] Array de entrada.Retorno: [ndarray] Índices de elementos que no son cero. Los índices se agrupan por elemento.
Código #1:
# Python program explaining
# argwhere() function
import numpy as geek
# input array
in_arr = [[ 2, 0, 7], [ 0, 5, 9]]
print ("Input array : ", in_arr)
out_arr = geek.argwhere(in_arr)
print ("Output indices of non zero array element: \n", out_arr)
Producción:
Input array : [[2, 0, 7], [0, 5, 9]] Output indices of non zero array element: [[0 0] [0 2] [1 1] [1 2]]
Código #2:
# Python program explaining
# argwhere() function
import numpy as geek
# input array
in_arr = geek.arange(8).reshape(4, 2)
print ("Input array : ", in_arr)
out_arr = geek.argwhere(in_arr>4)
print ("Output indices greater than 4: \n", out_arr)
Producción:
Input array : [[0 1] [2 3] [4 5] [6 7]] Output indices greater than 4: [[2 1] [3 0] [3 1]]
Publicación traducida automáticamente
Artículo escrito por jana_sayantan y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA