关键词

详解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]]])

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

本文链接:http://task.lmcjl.com/news/17117.html

展开阅读全文