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

import numpy as geek

array = [[ 5 , 6 , 2 , 7 , 1 ],

`` [ 4 , 9 , 2 , 9 , 3 ]]

print ( "Original array : \n" , array)

print ( "\nTaking Indices\n" , geek.take(array, [ 0 , 4 ]))

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.