1. stack
stack的两个对象必须形状相同,结果会向外扩展一个维度,例:
import numpy as np
a = np.array([[1,2,3,4], [2,3,4,5]])
b = np.array([[2,3,4,5],[5,6,7,8]])
print(a.shape, a)
print(b.shape, b)
c = np.stack((a,b), axis=0)
print(c.shape, c)
2. vstack
直接在行上合并,要求列数相同,不会拓展维度,例:
a = np.array([[1,2,3,4], [2,3,4,5], [3,4,5,6]])
b = np.array([[2,3,4,5],[5,6,7,8]])
print("a: ", a.shape, a)
print("b: ", b.shape, b)
c = np.vstack((a,b))
print("c: ", c.shape, c)
3. hstack
直接在列上合并,要求行数相同,不会拓展维度,例:
a = np.array([[1,2], [2,3], [3,4]])
b = np.array([[2,3,4,5],[5,6,7,8], [5,6,7,8]])
print("a: ", a.shape, a)
print("b: ", b.shape, b)
c = np.hstack((a,b))
print("c: ", c.shape, c)