Python | Numpy numpy.choose()

Con la ayuda del Numpy numpy.choose()método, podemos seleccionar los elementos de una array multidimensional pasando un parámetro como una array que contiene el índice del número de fila que se seleccionará. Array de salida que tiene el mismo tamaño que el pasado en el parámetro.

Sintaxis: numpy.choose()

Retorno: Devuelve una array de opciones de elementos .

Ejemplo #1:
En este ejemplo podemos ver que con la ayuda del numpy.choose()método podemos extraer una array de opciones de una array multidimensional.

# import the important module in python
import numpy as np
             
# make a matrix with numpy
gfg = [[1, 2, 3, 4], [3, 1, 5, 6]]
             
# applying numpy.choose() method
geeks = np.choose([1, 0, 1, 0], gfg)
       
print(geeks)

array([3 2 5 4])

Ejemplo #2:

# import the important module in python
import numpy as np
             
# make a matrix with numpy
gfg = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
             
# applying numpy.choose() method
geeks = np.choose([2, 0, 1], gfg)
       
print(geeks)

array([7 2 6])