numpy.take() in Python

0 0 1 minute read

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 method
 
import 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 = 1
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]]

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.zambiatek.com or mail your article to review-team@zambiatek.com. 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.