numpy.take() in Python (original) (raw)

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 `

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.