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 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)
|
Output:
array([3 2 5 4])
Example #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)
|
Output:
array([7 2 6])