详解Numpy stack()（沿着新的轴堆叠数组）函数的作用与使用方法

Numpy中的stack()函数可以将多个数组沿着指定的轴堆叠起来，生成一个新的多维数组。该函数主要有两个参数，第一个是待堆叠的数组，第二个是沿着哪个轴进行堆叠。常见的轴为0和1，分别表示沿着行和列进行堆叠。如果没有指定轴参数，则默认为0轴。

使用方法：

numpy.stack(arrays, axis=0)

参数解释：

• arrays：需要堆叠的多个数组。
• axis：沿着哪个轴进行堆叠，默认为0轴。

示例1：沿着列堆叠数组

假设有两个数组a和b，分别为：

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

b = [[7,8,9],
     [10,11,12]]

现在我们想要将这两个数组沿着列（即第2个轴）进行堆叠，可以使用以下代码：

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

这将会得到一个新的数组c，它的值为：

array([[[ 1,  2,  3],
        [ 7,  8,  9]],

       [[ 4,  5,  6],
        [10, 11, 12]]])

其中第一个轴代表行，第二个轴代表列，第三个轴代表单个数组中的元素。

示例2：沿着行堆叠数组

假设有两个数组a和b，分别为：

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

b = [[7,8,9],
     [10,11,12]]

现在我们想要将这两个数组沿着行（即第1个轴）进行堆叠，可以使用以下代码：

c = np.stack((a,b), axis=0)

这将会得到一个新的数组c，它的值为：

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

       [[ 7,  8,  9],
        [10, 11, 12]]])

其中第一个轴代表堆叠后的数组个数，第二个轴代表行，第三个轴代表单个数组中的元素。