数组的拼接

 1 #导入numpy
 2 import numpy as np 
 3 #创建两个数组
 4 a = np.array([[1,2,3],[4,5,6]])
 5 b = np.array([[11,12,13],[14,15,16]])
 6 print(a)
 7 print(b)
 8 #使用hstack进行水平拼接
 9 r = np.hstack([a,b])
10 rr = np.hstack((a,b))
11 print(r) 
12 print(rr)
13 #使用vstack进行垂直方向拼接
14 r = np.vstack((a,b))
15 print(r)
16 #使用concatenate
17 print('axis=0 默认情况 垂直方向拼接 相当于vstack')
18 r1 = np.concatenate((a,b),axis=0)
19 r2 = np.concatenate((a,b))
20 print(r1)
21 print(r2)
22 
23 #对于二维数组有两个轴 axis=0 axis=1
24 print('axis=1  水平方向拼接 相当于hstack')
25 r3 = np.concatenate((a,b),axis=1)
26 print(r3)
27 
28 #三维数组有三个轴 axis=0 1 2
29 a1 = np.arange(1,13).reshape(1,2,6)
30 print(a1,a1.shape)
31 a2 = np.arange(101,113).reshape(1,2,6)
32 print(a2,a2.shape)
33 print('三维 axis=0')
34 r1 = np.concatenate((a1,a2),axis=0)
35 print(r1,r1.shape)
36 print('三维 axis=1')
37 r2 = np.concatenate((a1,a2),axis=1)
38 print(r2,r2.shape)
39 print('三维 axis=2')
40 r3 = np.concatenate((a1,a2),axis=2)
41 print(r3,r3.shape)

 1 [[1 2 3]
 2  [4 5 6]]
 3 [[11 12 13]
 4  [14 15 16]]
 5 [[ 1  2  3 11 12 13]
 6  [ 4  5  6 14 15 16]]
 7 [[ 1  2  3 11 12 13]
 8  [ 4  5  6 14 15 16]]
 9 [[ 1  2  3]
10  [ 4  5  6]
11  [11 12 13]
12  [14 15 16]]
13 axis=0 默认情况 垂直方向拼接 相当于vstack
14 [[ 1  2  3]
15  [ 4  5  6]
16  [11 12 13]
17  [14 15 16]]
18 [[ 1  2  3]
19  [ 4  5  6]
20  [11 12 13]
21  [14 15 16]]
22 axis=1  水平方向拼接 相当于hstack
23 [[ 1  2  3 11 12 13]
24  [ 4  5  6 14 15 16]]
25 [[[ 1  2  3  4  5  6]
26   [ 7  8  9 10 11 12]]] (1, 2, 6)
27 [[[101 102 103 104 105 106]
28   [107 108 109 110 111 112]]] (1, 2, 6)
29 三维 axis=0
30 [[[  1   2   3   4   5   6]
31   [  7   8   9  10  11  12]]
32 
33  [[101 102 103 104 105 106]
34   [107 108 109 110 111 112]]] (2, 2, 6)
35 三维 axis=1
36 [[[  1   2   3   4   5   6]
37   [  7   8   9  10  11  12]
38   [101 102 103 104 105 106]
39   [107 108 109 110 111 112]]] (1, 4, 6)
40 三维 axis=2
41 [[[  1   2   3   4   5   6 101 102 103 104 105 106]
42   [  7   8   9  10  11  12 107 108 109 110 111 112]]] (1, 2, 12)