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.
Share your thoughts in the comments
Please Login to comment...