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

stack()函数
函数原型为：stack(arrays,axis=0)
主要的参数有两个

一个是arrays，也就是用来作为堆叠的数组，要求形状维度必须相等
第二个参数为axis也就是指定依照哪个维度进行堆叠（axis默认是0）
理解了axis的用法也就能理解是根据什么来进行堆叠了，找到要堆叠的元素的维度，然后对应位置的元素堆叠起来即可

>>> a=[[1,2,3],
...    [4,5,6]]
>>> a
[[1, 2, 3], [4, 5, 6]]
>>> a[1][1]
5
>>> c=np.stack(a,axis=0) #axis=0表示将最外层[ ]里，只有两个元素[1, 2, 3],[4, 5, 6]，直接堆叠
>>> c
array([[1, 2, 3],
       [4, 5, 6]])
>>> c[1][1]
5
>>> c=np.stack(a,axis=1) #axis=1则看第二层的[ ]，每个第二层的[ ]都有三个元素1,2,3和4,5,6，因此对应元素就行堆叠
>>> c
array([[1, 4],
       [2, 5],
       [3, 6]])
       
>>> type(c)
<class 'numpy.ndarray'>


三维情况
>>> a=[[1,2,3],
...    [4,5,6]]
>>> b=[[1,2,3],
...    [4,5,6]]
>>> c=[[1,2,3],
...    [4,5,6]]
>>> a
[[1, 2, 3], [4, 5, 6]]
>>> b
[[1, 2, 3], [4, 5, 6]]
>>> c
[[1, 2, 3], [4, 5, 6]]
>>> d=np.stack((a,b,c),axis=0)
>>> d
array([[[1, 2, 3],
        [4, 5, 6]],

       [[1, 2, 3],
        [4, 5, 6]],

       [[1, 2, 3],
        [4, 5, 6]]])
>>> d=np.stack((a,b,c),axis=1)
>>> d
array([[[1, 2, 3],
        [1, 2, 3],
        [1, 2, 3]],

       [[4, 5, 6],
        [4, 5, 6],
        [4, 5, 6]]])
>>> d=np.stack((a,b,c),axis=2)
>>> d
array([[[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3]],

       [[4, 4, 4],
        [5, 5, 5],
        [6, 6, 6]]])