Skip to content
Related Articles

Related Articles

Python | Numpy numpy.choose()

Improve Article
Save Article
  • Last Updated : 10 Apr, 2019
Improve Article
Save Article

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])

My Personal Notes arrow_drop_up
Related Articles

Start Your Coding Journey Now!