Skip to content
Related Articles

Related Articles

Improve Article

Python | Numpy matrix.take()

  • Last Updated : 29 May, 2019

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

 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




My Personal Notes arrow_drop_up
Recommended Articles
Page :