Python | Método Numpy np.digitize()

Con la ayuda del np.digitize()método, podemos obtener los índices de los contenedores a los que pertenece cada valor de una array utilizando el np.digitize()método.

Sintaxis: np.digitize(Array, Bin, Right)
Retorno: Retorna una array de índices de los contenedores.

Ejemplo #1:
En este ejemplo, podemos ver que al usar el np.digitize()método, podemos obtener la array de índices del contenedor de cada valor que pertenece a una array al usar este método.

# import numpy
import numpy as np
  
a = np.array([1.2, 2.4, 3.6, 4.8])
bins = np.array([1.0, 1.3, 2.5, 4.0, 10.0])
  
# using np.digitize() method
gfg = np.digitize(a, bins)
  
print(gfg)

Producción :

[1 2 3 4]

Ejemplo #2:

# import numpy
import numpy as np
  
a = np.array([[10.2, 21.4, 3.6, 14.8], [1.0, 5.0, 10.0, 15.0]])
bins = np.array([1.0, 1.3, 2.5, 4.0, 10.0])
  
# using np.digitize() method
gfg = np.digitize(a, bins)
  
print(gfg)

Producción :

[[5 5 3 5]
[1 4 5 5]]