función numpy.select() | Python

numpy.select()()La función devuelve una array extraída de elementos en la lista de opciones, según las condiciones.

Sintaxis: numpy.select(condlist, choicelist, default = 0)
Parámetros:
condlist: [lista de bool ndarrays] Determina de qué array en la lista de opciones se toman los elementos de salida. Cuando se cumplen varias condiciones, se utiliza la primera que se encuentra en condlist.
choicelist: [lista de ndarrays] La lista de arrays de las que se toman los elementos de salida. Tiene que tener la misma longitud que condlist.
predeterminado: [escalar, opcional] El elemento insertado en la salida cuando todas las condiciones se evalúan como Falso.
Retorno: [ndarray] Una array extraída de elementos en la lista de opciones, según las condiciones.

Código #1:

# Python program explaining
# numpy.select() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.arange(8)
  
condlist = [arr<3, arr>4]
choicelist = [arr, arr**3]
  
gfg = geek.select(condlist, choicelist)
  
print (gfg)

Producción :

[  0, 1, 2, 0, 0, 125, 216, 343]

 
Código #2:

# Python program explaining
# numpy.select() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.arange(8)
  
condlist = [arr<4, arr>6]
choicelist = [arr, arr**2]
  
gfg = geek.select(condlist, choicelist)
  
print (gfg)

Producción :

[ 0, 1, 2, 3, 0, 0, 0, 49]