📜  Python中的numpy.take

📅  最后修改于: 2020-06-05 04:24:03             🧑  作者: Mango

numpy.take(array, indices, axis = None, out = None, mode =’raise’) : 沿着提到的轴和索引从数组返回elememts。

参数:

array:array_like,输入数组
index:要获取的值的索引
axis:[int,可选]我们需要在其上获取元素的轴;
                  默认情况下[axis = None],使用拼合的输入
模式:[{'raise','wrap','clip'},可选]提到越界索引的行为
                  提高:[默认]引发错误
                  环绕:环绕
                  剪辑:剪辑到范围
out:[ndarray,可选]将结果放入数组

返回值:

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("原始数组 : \n", array) 
  
# indices = [0, 4] 
print("\n取索引\n", geek.take(array, [0, 4])) 
  
# indices = [0, 4] with axis = 1 
print("\n取索引\n", geek.take(array, [0, 4], axis = 1)) 

输出:

原始数组:
 [[5、6、2、7、1],[ 

4、9、2、9、3 ]] 

取索引
 [5 1] 

取索引
 [[5 1] 
 [4 3]]