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