Skip to content
Related Articles

Related Articles

Improve Article

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

 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning - Basic Level Course

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
Recommended Articles
Page :