numpy.take() in Python

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

Note : 
These codes won’t run on online IDE’s. So please, run them on your systems to explore the working.