Open In App

Python | Numpy matrix.take()

Improve
Improve
Like Article
Like
Save
Share
Report

With the help of Numpy matrix.take() method, we can select the elements from a given matrix by passing the parameter as index value of that element. It will return a matrix having one dimension. Remember it will work for one axis at a time.

Syntax : matrix.take(index, axis)
Return : Return matrix of selected indexes

Example #1 :
In this example we can see that by selecting one index we get only one value in a matrix by using matrix.take() method.




# import the important module in python
import numpy as np
              
# make matrix with numpy
gfg = np.matrix('[4, 1, 12, 3, 4, 6, 7]')
              
# applying matrix.take() method
geek = gfg.take(2)
    
print(geek)


Output:

[[12]]

Example #2 :




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


Output:

[[ 4 12  4]]

Last Updated : 29 May, 2019
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads