np.stack()我的理解是堆叠数据,把不同维度的数据组合起来。
代码如下:
#生成数据
a=np.zeros((3,4))
b=np.ones((3,4))
c=np.array(b+1)
#堆叠数组
s0=np.stack([a,b,c],axis=0) #第0维,堆叠整个数组 array
s1=np.stack([a,b,c],axis=1) #第1维,堆叠 数组第1维的数据 array[]
s2=np.stack([a,b,c],axis=2) #第2维,堆叠 数组第2维的数据 array[][]
s3=np.stack([a,b,c],axis=-1) #第-1维,堆叠 数组第-1维的数据,就是最里面的数
结果如下:
a
[[0. 0. 0. 0.]
[0. 0. 0. 0.]
[0. 0. 0. 0.]]
----------------------
b
[[1. 1. 1. 1.]
[1. 1. 1. 1.]
[1. 1. 1. 1.]]
----------------------
c
[[2. 2. 2. 2.]
[2. 2. 2. 2.]
[2. 2. 2. 2.]]
----------------------
axis=0
[[[0. 0. 0. 0.]
[0. 0. 0. 0.]
[0. 0. 0. 0.]]
[[1. 1. 1. 1.]
[1. 1. 1. 1.]
[1. 1. 1. 1.]]
[[2. 2. 2. 2.]
[2. 2. 2. 2.]
[2. 2. 2. 2.]]]
----------------------
axis=1
[[[0. 0. 0. 0.]
[1. 1. 1. 1.]
[2. 2. 2. 2.]]
[[0. 0.