numpy.select()() function return an array drawn from elements in choicelist, depending on conditions.
Syntax : numpy.select(condlist, choicelist, default = 0)
Parameters :
condlist : [list of bool ndarrays] It determine from which array in choicelist the output elements are taken. When multiple conditions are satisfied, the first one encountered in condlist is used.
choicelist : [list of ndarrays] The list of arrays from which the output elements are taken. It has to be of the same length as condlist.
default : [scalar, optional] The element inserted in output when all conditions evaluate to False.
Return : [ndarray] An array drawn from elements in choicelist, depending on conditions.
Code #1 :
| # Python program explaining# numpy.select() function Â# importing numpy as geek importnumpy as geek Âarr =geek.arange(8) Âcondlist =[arr<3, arr>4]choicelist =[arr, arr**3] Âgfg =geek.select(condlist, choicelist) Âprint(gfg) | 
Output :
[ 0, 1, 2, 0, 0, 125, 216, 343]
 
Code #2 :
| # Python program explaining# numpy.select() function Â# importing numpy as geek importnumpy as geek Âarr =geek.arange(8) Âcondlist =[arr<4, arr>6]choicelist =[arr, arr**2] Âgfg =geek.select(condlist, choicelist) Âprint(gfg) | 
Output :
[ 0, 1, 2, 3, 0, 0, 0, 49]


 
                                    







