python numpy 数组堆叠

numpy数组堆叠函数

stack(arrays, axis) : 沿着新轴连接数组的序列

column_stack(): 将1维数组作为列堆叠到2维数组中

hstack(): 按水平方向堆叠数组

vstack(): 按垂直方向堆叠数组

具体应用如下代码

import numpy as np
a = np.array([1,2,3])
b = np.array([4,5,6])
c = np.array([[11],[22],[33]])
d = np.array([[44],[55],[66]])

1. stack(arrays, axis)

stack(arrays, axis=-1) 当axis = -1时，表示沿水平方向堆叠

np.stack((a,b),axis=-1)

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

从结果看，就是把a,b作为新数组的一列堆叠在一起

stack(arrays, axis=1) 当axis = 0 时，表示沿垂直方向堆叠

np.stack((a,b),axis=0)

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

2. column_stack()

np.column_stack((a,b))

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

np.column_stack((c,d))

array([[11, 44],
       [22, 55],
       [33, 66]])

3 hstack()

np.hstack((a,b))

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

np.hstack((c,d))

array([[11, 44],
       [22, 55],
       [33, 66]])

4. vstack()

np.vstack((a,b))

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

np.vstack((c,d))

array([[11],
       [22],
       [33],
       [44],
       [55],
       [66]])