沿轴取数组中的元素。
这个函数与“花式”索引(使用数组索引数组)的功能相同; 但是,如果您需要沿给定轴的元素,则可以更容易使用。
参数: |
a:array_like
indices:array_like
axis:int,可选
out:ndarray,可选
mode:{‘raise‘,‘wrap‘,‘clip‘},可选
|
---|---|
返回: |
子阵:ndarray
|
a = np.array([[1, 2],
[3, 4]])
b = np.take(a,[0,1])
print(b)
>>[1 2]
a = np.array([[1, 2],
[3, 4]])
b = np.take(a,0,axis=0)
print(b)
>>[1 2]
a = np.array([[1, 2],
[3, 4]])
b = np.take(a,1,axis=1)
print(b)
>>[2 4]
a = np.array([[1, 2],
[3, 4]])
b = np.take(a,[0,1],axis=1)
print(b)
>>[[1,2]
[3,4]]
常用参数:
indices:参数代表你要截取数据的索引
axis:代表你要怎么截取,是按行截取还是按列截取,axis=1,竖着截取,axis=0横着截取
原文:https://www.cnblogs.com/daguonice/p/11389078.html