Numpy中的stack()函数可以将多个数组沿着指定的轴堆叠起来,生成一个新的多维数组。该函数主要有两个参数,第一个是待堆叠的数组,第二个是沿着哪个轴进行堆叠。常见的轴为0和1,分别表示沿着行和列进行堆叠。如果没有指定轴参数,则默认为0轴。
使用方法:
numpy.stack(arrays, axis=0)
参数解释:
假设有两个数组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]]])
其中第一个轴代表行,第二个轴代表列,第三个轴代表单个数组中的元素。
假设有两个数组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