np.stack()使用

1.官网
https://numpy.org/doc/stable/reference/generated/numpy.stack.html

np.stack(arrays , axis =0)：沿新轴加入一系列数组。

1.参数

• arrays：（必填）一序列像数组的数，每一个数组都有相同的大小
• axis ：（可选）结果数组中的轴，输入数组沿该轴堆叠。
• out：（可选）如果提供，则放置结果的目的地。

2.返回值

stacked：ndarray
返回值必输入数组多一个维度。

3.示例

以二维为例，更高维的还没理解。

a = np.ones([2,3])
b = np.zeros([2,3])
c = np.arange(1,7).reshape(2,3)
list =[]
list.append(a)
list.append(b)
list.append(c)
print(a)
print(b)
print(c)
#print(np.stack((a,b),axis=2))
print(np.stack(list,axis=0),np.stack(list,axis=0).shape)
print(np.stack(list,axis=1),np.stack(list,axis=1).shape)
print(np.stack(list,axis=2),np.stack(list,axis=2).shape)

axis=0，平铺排列

axis=1，每一个数组的第一行横着排

axis=2，每一个数组的每一行，竖着排