Numpy——stack()、hstack()、vstack()、dstack()

这四个函数的作用都是将几个数组进行合并，不同的是合并的方式有所不同。

stack()

函数原型为：stack(arrays, axis=0)，arrays可以传数组、列表、元组。

import numpy as np
a = [1,2,3]
b = [4,5,6]
np.stack((a,b),axis=0)
array([[1, 2, 3],
       [4, 5, 6]])
np.stack((a,b),axis=1)
array([[1, 4],
       [2, 5],
       [3, 6]])

vstack()

函数原型：vstack(tup) ，参数tup可以是元组，列表，或者numpy数组，返回结果为numpy的数组。
功能：实现了轴0合并，合并后的轴0为各个子元素的轴0相加。

a = np.array([[1,2,3]]) #（1，3）
b = np.array([[4,5,6]]) #（1，3）
c = np.vstack((a,b))  #（2，3）
a = np.array([[1],[2],[3]]) #(3, 1)
b = np.array([[4],[5],[6]])  #(3, 1)
c = np.vstack((a,b)) #(6，1)

hstack()

函数原型：hstack(tup) ，参数tup可以是元组，列表，或者numpy数组，返回结果为numpy的数组。
功能：实现了轴1合并，合并后的轴1为各个子元素的轴1相加。

a = np.array([[1,2,3]]) #（1，3）
b = np.array([[4,5,6]]) #（1，3）
c = np.hstack((a,b))  #（1，6）
a = np.array([[1],[2],[3]]) #(3, 1)
b = np.array([[4],[5],[6]])  #(3, 1)
c = np.hstack((a,b)) #(3, 2)

dstack()

dstack自然是说deep stack了，它是轴2合并。dstack([a,b])把a和b摞在一起，像一摞扑克牌一样。

a = np.array([1,2,3]) #（3,1）
b = np.array([4,5,6]) #（3,1）
c = np.dstack((a,b))  #（1，3,2）
array([[[1, 4],
        [2, 5],
        [3, 6]]])