python多维数组切片方法

codepython 2018-04-13

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

相关推荐