numpy.take() in Python
Last Updated :
08 Mar, 2024
The numpy.take() function returns elements from array along the mentioned axis and indices.
Syntax: numpy.take(array, indices, axis = None, out = None, mode ='raise')
Parameters :
array : array_like, input array
indices : index of the values to be fetched
axis : [int, optional] axis over which we need to fetch the elements;
By Default[axis = None], flattened input is used
mode : [{‘raise’, ‘wrap’, ‘clip’}, optional] mentions how out-of-bound indices will behave
raise : [default]raise an error
wrap : wrap around
clip : clip to the range
out : [ndarray, optional]to place result within array
Returns :
ndarray; array has the same type
Python
import numpy as geek
array = [[ 5 , 6 , 2 , 7 , 1 ],
[ 4 , 9 , 2 , 9 , 3 ]]
print ( "Original array : \n" , array)
print ( "\nTaking Indices\n" , geek.take(array, [ 0 , 4 ]))
print ( "\nTaking Indices\n" , geek.take(array, [ 0 , 4 ], axis = 1 ))
|
Output :
Original array :
[[5, 6, 2, 7, 1], [4, 9, 2, 9, 3]]
Taking Indices
[5 1]
Taking Indices
[[5 1]
[4 3]]
Note :
These codes won’t run on online IDE’s. So please, run them on your systems to explore the working.
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...