Python | Numpy matrix.choose()

With the help of Numpy matrix.choose() method, we can select the elements from a matrix 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 : matrix.choose()

Return : Return an array of element choice



Example #1 :
In this example we can see that with the help of matrix.choose() method we are able to extract an array of choices from matrix.

filter_none

edit
close

play_arrow

link
brightness_4
code

# import the important module in python
import numpy as np
             
# make a matrix with numpy
gfg = np.matrix('[1, 2, 3, 4; 3, 1, 5, 6]')
             
# applying matrix.choose() method
geeks = np.choose([1, 0, 1, 0], gfg)
       
print(geeks)

chevron_right


Output:

[[3 2 5 4]]

Example #2 :

filter_none

edit
close

play_arrow

link
brightness_4
code

# import the important module in python
import numpy as np
             
# make a matrix with numpy
gfg = np.matrix('[1, 2, 3; 4, 5, 6; 7, 8, 9]')
             
# applying matrix.choose() method
geeks = np.choose([2, 0, 1], gfg)
       
print(geeks)

chevron_right


Output:

[[7 2 6]]


My Personal Notes arrow_drop_up

Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.