Python | Numpy numpy.choose()
Last Updated :
10 Apr, 2019
With the help of Numpy numpy.choose()
method, we can select the elements from an multidimensional array by passing a parameter as an array which contain the index of row number to be selected. Output array having the same size as passed in the parameter.
Syntax : numpy.choose()
Return : Return an array of element choice
Example #1 :
In this example we can see that with the help of numpy.choose()
method we are able to extract an array of choices from multidimensional array.
import numpy as np
gfg = [[ 1 , 2 , 3 , 4 ], [ 3 , 1 , 5 , 6 ]]
geeks = np.choose([ 1 , 0 , 1 , 0 ], gfg)
print (geeks)
|
Example #2 :
import numpy as np
gfg = [[ 1 , 2 , 3 ], [ 4 , 5 , 6 ], [ 7 , 8 , 9 ]]
geeks = np.choose([ 2 , 0 , 1 ], gfg)
print (geeks)
|
Share your thoughts in the comments
Please Login to comment...