numpy中数组切片

二维数组

import numpy as np 

a = np.array([[1,2,3],[4,5,6]])
print(a.shape)

print(a[0,:],a[0,:].shape)
print(a[-1,:],a[-1,:].shape)
print(a[0:2,:],a[0:2,:].shape)
print("------------------")
print(a[:,0],a[:,0].shape)
print(a[:,1],a[:,1].shape)
print(a[:,0:2],a[:,0:2].shape)

(2, 3)
[1 2 3] (3,)
[4 5 6] (3,)
[[1 2 3]
 [4 5 6]] (2, 3)
------------------
[1 4] (2,)
[2 5] (2,)
[[1 2]
 [4 5]] (2, 2)

三维数组

import numpy as np 

b = np.array([[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]],
              [[13, 14, 15, 16], [17, 18, 19, 20], [21, 22, 23, 24]],
              [[25, 26, 27, 28], [29, 30, 31, 32], [33, 34, 35, 36]],
              ])
print(b.shape)

print(b[0,::],b[0,::].shape)
print(b[1,::],b[1,::].shape)
print(b[0:2,::],b[0:2,::].shape)

(3, 3, 4)
[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]] (3, 4)
[[13 14 15 16]
 [17 18 19 20]
 [21 22 23 24]] (3, 4)
[[[ 1  2  3  4]
  [ 5  6  7  8]
  [ 9 10 11 12]]

 [[13 14 15 16]
  [17 18 19 20]
  [21 22 23 24]]] (2, 3, 4)

print(b[::,0],b[::,0].shape)
print(b[::,1],b[::,1].shape)
print(b[::,0:2],b[::,0:2].shape)

[[ 1  2  3  4]
 [13 14 15 16]
 [25 26 27 28]] (3, 4)
[[ 5  6  7  8]
 [17 18 19 20]
 [29 30 31 32]] (3, 4)
[[[ 1  2  3  4]
  [ 5  6  7  8]]

 [[13 14 15 16]
  [17 18 19 20]]

 [[25 26 27 28]
  [29 30 31 32]]] (3, 2, 4)

print(b[:, 0:], b[:, 0:].shape)
print(b[:, 1:], b[:, 1:].shape)
print(b[:, -1:], b[:, -1:].shape)
print(b[:, 0:2:], b[:, 0:2:].shape)

[[[ 1  2  3  4]
  [ 5  6  7  8]
  [ 9 10 11 12]]

 [[13 14 15 16]
  [17 18 19 20]
  [21 22 23 24]]

 [[25 26 27 28]
  [29 30 31 32]
  [33 34 35 36]]] (3, 3, 4)
[[[ 5  6  7  8]
  [ 9 10 11 12]]

 [[17 18 19 20]
  [21 22 23 24]]

 [[29 30 31 32]
  [33 34 35 36]]] (3, 2, 4)
[[[ 9 10 11 12]]

 [[21 22 23 24]]

 [[33 34 35 36]]] (3, 1, 4)
[[[ 1  2  3  4]
  [ 5  6  7  8]]

 [[13 14 15 16]
  [17 18 19 20]]

 [[25 26 27 28]
  [29 30 31 32]]] (3, 2, 4)

print(b[:, :, 0], b[:, :, 0].shape)
print(b[:, :, 1], b[:, :, 1].shape)
print(b[:, :, -1], b[:, :, -1].shape)
print(b[:, :, 0:2:], b[:, :, 0:2].shape)

[[ 1  5  9]
 [13 17 21]
 [25 29 33]] (3, 3)
[[ 2  6 10]
 [14 18 22]
 [26 30 34]] (3, 3)
[[ 4  8 12]
 [16 20 24]
 [28 32 36]] (3, 3)
[[[ 1  2]
  [ 5  6]
  [ 9 10]]

 [[13 14]
  [17 18]
  [21 22]]

 [[25 26]
  [29 30]
  [33 34]]] (3, 3, 2)