python多维数组切片方法

2023年5月11日 203次阅读

1、数组a第0个元素（二维数组）下的所有子元素（一维数组）的第一列

import numpy as np
b=np.arange(24)
a=b.reshape(2,3,4)
print a
print a[0,:,0]

2、取所有二维数组下的每个二维数组的第0个元素（一维数组）

b=np.arange(24)
a=b.reshape(2,3,4)
print a
print '--------------------'
print a[:,0]

结果：

[[ 0 1 2 3]
 [12 13 14 15]]

3、取所有二维数组下面的所有一维数组的每一个一维数组的第0个元素

b=np.arange(24)
a=b.reshape(2,3,4)
print a
print '--------------------'
print a[:,:,0]

结果：

[[ 0 4 8]
 [12 16 20]]

4、取所有二维数组下面的所有一维数组的每一个一维数组的第0到2（不包含第2）个元素

b=np.arange(24)
a=b.reshape(2,3,4)
print a
print '--------------------'
print a[:,:,0:2]

结果：

[[[ 0 1]
 [ 4 5]
 [ 8 9]]
 [[12 13]
 [16 17]
 [20 21]]]

点赞