Python numpy.take()

Python numpy.take()函数返回数组中沿上述axis和下标方向的元素。

语法: numpy.take(array, indices, axis = None, out = None, mode =’raise’)

参数 :

array : array_like, input array

indices : 要获取的值的索引
axis : [int, optional] 我们需要获取元素的轴；根据默认[axis = None]，使用扁平化的输入。
mode : [{‘raise’, ‘wrap’, ‘clip’}, optional] 提到了边界外的指数将如何表现
raise : [default]引发一个错误
wrap : 绕开
clip : 夹到范围内
out : [ndarray, optional]将结果放在数组中。

返回值 :

ndarray; 数组具有相同的类型

示例：

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