numpy 数组合并

import numpy as np

创建数组

a1 = np.arange(0,6).reshape(2,3)
a1

array([[0, 1, 2],
[3, 4, 5]])

a2 = np.arange(6,12).reshape(2,3)
a2

array([[ 6, 7, 8],
[ 9, 10, 11]])

append函数

np.append(a1,a2)

array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])

np.append(a1,a2,axis=0)

array([[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8],
[ 9, 10, 11]])

np.append(a1,a2,axis=1)

array([[ 0, 1, 2, 6, 7, 8],
[ 3, 4, 5, 9, 10, 11]])

concatenate函数

np.concatenate((a1,a2),axis=0)

array([[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8],
[ 9, 10, 11]])

np.concatenate((a1,a2),axis=1)

array([[ 0, 1, 2, 6, 7, 8],
[ 3, 4, 5, 9, 10, 11]])

stack函数

np.vstack((a1,a2))

array([[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8],
[ 9, 10, 11]])

np.hstack((a1,a2))

array([[ 0, 1, 2, 6, 7, 8],
[ 3, 4, 5, 9, 10, 11]])

np.dstack((a1,a2))

array([[[ 0, 6],
[ 1, 7],
[ 2, 8]],

​ [[ 3, 9],
​ [ 4, 10],
​ [ 5, 11]]])