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
# Python Program illustrating# numpy.take methodimport numpy as geek#array = geek.arange(10).reshape(2, 5)array = [[5, 6, 2, 7, 1], [4, 9, 2, 9, 3]]print("Original array : \n", array)# indices = [0, 4]print("\nTaking Indices\n", geek.take(array, [0, 4]))# indices = [0, 4] with axis = 1print("\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]]
References :
https://docs.scipy.org/doc/numpy-dev/reference/generated/numpy.take.html#numpy.take
Note :
These codes won’t run on online IDE’s. So please, run them on your systems to explore the working.
This article is contributed by Mohit Gupta_OMG 😀. If you like Lazyroar and would like to contribute, you can also write an article using write.geeksforgeeks.org or mail your article to review-team@geeksforgeeks.org. See your article appearing on the Lazyroar main page and help other Geeks.
Please write comments if you find anything incorrect, or you want to share more information about the topic discussed above.
