GeeksforGeeks App
Open App
Browser
Continue

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

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 GeeksforGeeks and would like to contribute, you can also write an article using write.geeksforgeeks.org or mail your article to review-team@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.