Python中的 numpy.take()

numpy.take()函数沿提到的轴和索引从数组中返回元素。

Syntax: numpy.take(array, indices, axis = None, out = None, mode ='raise')

参数 ：

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

返回：

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

输出 ：

Original array : 
 [[5, 6, 2, 7, 1], [4, 9, 2, 9, 3]]

Taking Indices
 [5 1]

Taking Indices
 [[5 1]
 [4 3]]